nn.MaxPool2d()、transpose().contiguous()、view()说明

发布时间 2023-07-26 22:39:42作者: 孜孜不倦fly

1.nn.MaxPool2d()
和nn.Conv2D()基本一样,但是stride默认值是kernel_size。
2.transpose().contiguous()、view()
contiguous一般与transpose,permute,view搭配使用:使用transpose或permute进行维度变换后,调用contiguous,然后方可使用view对维度进行变形

print(x.shape)
x = x.transpose(1, 3).contiguous()
print(x.shape)
x = x.view(x.size(0), -1, x.size(3))
print(x.shape)

输出

torch.Size([8, 128, 40, 256])
torch.Size([8, 256, 40, 128])
torch.Size([8, 10240, 128])

参考网站:pytorch中的contiguous()函数的浅浅解释