在 PyTorch 中,contiguous()
是用于检查张量是否是按照内存中的连续顺序存储的函数。如果张量是按照内存中的连续顺序存储的,则返回原始张量;否则将会创建一个新的连续张量,并将数据复制到该张量中。
假设有一个形状为 (3, 4, 5) 的三维张量 x
,其中每个元素都是从 0 开始的整数:
import torch
x = torch.arange(60).reshape(3, 4, 5)
print(x)
输出
在 PyTorch 中,contiguous()
是用于检查张量是否是按照内存中的连续顺序存储的函数。如果张量是按照内存中的连续顺序存储的,则返回原始张量;否则将会创建一个新的连续张量,并将数据复制到该张量中。
假设有一个形状为 (3, 4, 5) 的三维张量 x
,其中每个元素都是从 0 开始的整数:
import torch
x = torch.arange(60).reshape(3, 4, 5)
print(x)
输出