pytorch方法測試 卷積(二維)

2021-09-07 14:08:24 字數 897 閱讀 1867

pytorch方法測試——卷積(二維)

測試**:

import torch

import torch.nn as nn

m = nn.conv2d(2, 2, 3, stride=2)

input = torch.randn(1, 2, 5, 7)

output = m(input)

print(input)

print("卷積的權重:")

print(m.weight)

print("卷積的偏重:")

print(m.bias)

print("二維卷積後的輸出:")

print(output)

print("輸出的尺度:")

print(output.size())

convblockone = 0

convblocktwo = 0

for i in range(3):

for j in range(3):

# 第乙個卷積核與對應相乘

convblockone += m.weight[0][0][i][j] * input[0][0][i][j] \

+ m.weight[0][1][i][j] * input[0][1][i][j]

# 第二個卷積核與對應相乘

convblocktwo += m.weight[1][0][i][j] * input[0][0][i][j] \

+ m.weight[1][1][i][j] * input[0][1][i][j]

convblockone += m.bias[0]

convblocktwo += m.bias[1]

print("第

二維卷積詳細解釋

其中,矩陣a和b的尺寸分別為ma na即mb nb 對矩陣a補零,第一行之前和最後一行之後都補mb 1行,第一列之前和最後一列之後都補nb 1列 注意conv2不支援其他的邊界補充選項,函式內部對輸入總是補零 之所以都是 1是因為卷積核要在影象a上面移動,移動的時候需要滿足兩者至少有一列或者一行是重...

5 1 二維卷積層

卷積神經網路 convolutional neural network 是含有卷積層 convolutional layer 的神經網路 二維互相關運算 通常在卷積層中使用更加直觀的互相關 cross correlation 運算,在二維卷積層中,乙個二維輸入陣列和乙個二維核 kernel 陣列通過...

二維矩陣與二維矩陣之間的卷積

最近在學習數字影象處理 digital image processing,dip 這門課,感覺有些吃力。由於在數字訊號處理 digital singal processing,dsp 這門課中只學了一維矩陣之間的卷積運算。假設我們的卷積核h為kernel矩陣 33 待處理矩陣f x,y 為 55 h...