torch.contiguous()
时间: 2023-04-12 13:04:40 浏览: 187
pytorch torch.expand和torch.repeat的区别详解
torch.contiguous()是一个Tensor方法,用于返回一个连续的Tensor,即在内存中连续存储的Tensor。如果Tensor不是连续的,该方法会返回一个新的Tensor,该Tensor的数据是连续的,但是与原来的Tensor不共享存储空间。该方法通常用于提高Tensor的计算效率。
阅读全文
pytorch torch.expand和torch.repeat的区别详解