permute函数将tensor的维度换位
contiguous()一般在permute()等改变形状和计算返回的tensor后面,因为改变形状后,有的tensor并不是占用一整块内存,而是由不同的数据块组成,而tensor的view()操作依赖于内存是整块的,这时只需要执行contiguous()这个函数,把tensor变成在内存中连续分布的形式。
本文详细解析了PyTorch中permute函数的功能,即如何通过该函数实现tensor维度的重新排列,并解释了contiguous函数的作用,尤其是在执行view操作前,如何确保tensor在内存中连续分布,以便顺利进行后续的计算和操作。
permute函数将tensor的维度换位
contiguous()一般在permute()等改变形状和计算返回的tensor后面,因为改变形状后,有的tensor并不是占用一整块内存,而是由不同的数据块组成,而tensor的view()操作依赖于内存是整块的,这时只需要执行contiguous()这个函数,把tensor变成在内存中连续分布的形式。
5557
655

被折叠的 条评论
为什么被折叠?