墨客（志锐）:
[code=python]
//验证使用
import torch.nn as nn
import torch
conv = nn.Conv2d(2,4,3, groups=2)
conv.weight.data = torch.tensor(
[[[[1.,1.,1.],
[1.,1.,1.],
[1.,1.,1.]]],
[[[2.,2.,2.],
[2.,2.,2.],
[2.,2.,2.]]],
[[[3.,3.,3.],
[3.,3.,3.],
[3.,3.,3.]]],
[[[4.,4.,4.],
[4.,4.,4.],
[4.,4.,4.]]]
])
conv.weight.requires_grad = False
print('conv 权重形状大小')
print(conv.weight.size())
x = torch.tensor(
[[[[1.,1.,1.,1.,1.],
[1.,1.,1.,1.,1.],
[1.,1.,1.,1.,1.],
[1.,1.,1.,1.,1.],
[1.,1.,1.,1.,1.]],
[[2.,2.,2.,2.,2.],
[2.,2.,2.,2.,2.],
[2.,2.,2.,2.,2.],
[2.,2.,2.,2.,2.],
[2.,2.,2.,2.,2.]]]])
print('输入形状大小')
print(x.size())
out = conv(x)
print('输出形状大小')
print(out)
[/code]