Pytorch: permute()函数,contiguous(),view()

permute函数将tensor的维度换位

contiguous()一般在permute()等改变形状和计算返回的tensor后面,因为改变形状后,有的tensor并不是占用一整块内存,而是由不同的数据块组成,而tensor的view()操作依赖于内存是整块的,这时只需要执行contiguous()这个函数,把tensor变成在内存中连续分布的形式。

https://blog.csdn.net/appleml/article/details/80143212

https://blog.csdn.net/york1996/article/details/81876886

你可能感兴趣的:(Pytorch)