PyTorch Code Snippets

1. 卷积计算

代码如下：

import torch
filter = torch.nn.Conv2d(2, 1, 2, 1, 1, 3, bias=False)

filter.weight[0][0][0][0]=2
filter.weight[0][0][0][1]=1
filter.weight[0][0][1][0]=1
filter.weight[0][0][1][1]=-1

filter.weight[0][1][0][0]=-3
filter.weight[0][1][0][1]=2
filter.weight[0][1][1][0]=-2
filter.weight[0][1][1][1]=3

x = torch.FloatTensor([[[[4,-3,2,2],[3,5,-2,3],[0,-4,1,7],[-8,2,9,2]], [[1,-5,3,7],[3,6,9,-2],[2,4,2,-1],[2,-2,7,0]]]])

filter(x)