pytorch方法测试——卷积（二维）

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("输入图片(2张)：")
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 +=