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 += 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("第
  
 

文章来源: blog.csdn.net，作者：网奇，版权归原作者所有，如需转载，请联系作者。

原文链接：blog.csdn.net/jacke121/article/details/85929638