卷积层权重的形状

conv = torch.nn.Conv2d(180, 2, (3, 3), stride=1, padding=1)
print(conv.weight.data.shape) 
#卷积核权重是一个四维张量，第一个维度表示卷积核个数，第二个维度表示每个卷积核的层数，第三和第四个维度表示卷积核的宽

输出结果为：    torch.Size([2, 180, 3, 3])