pytorch contiguous
时间: 2023-04-18 17:00:29 浏览: 209
pytorch
PyTorch中的contiguous函数是用来将Tensor变成连续的Tensor的。在PyTorch中,Tensor的存储方式是按照列优先(column-major)的方式存储的,也就是说,Tensor的每一列都是连续存储的,而每一行则不一定是连续存储的。当我们对一个Tensor进行操作时,如果它的存储方式不是连续的,那么就会导致计算效率的下降。因此,我们可以使用contiguous函数将Tensor变成连续的Tensor,从而提高计算效率。
阅读全文