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