torch.reshape如何使用

torch.reshape函数可以用来改变tensor的形状，其语法为torch.reshape(tensor, shape)，其中tensor为需要改变形状的tensor，shape为新的形状。使用时需要注意，新的形状必须与原来的形状元素个数相同，否则会报错。另外，reshape操作不会改变tensor的存储顺序，因此需要保证tensor是连续性的(contiguous)。如果不是连续性的，可以使用tensor.contiguous()函数将其转换为连续性的tensor。下面是一个使用torch.reshape函数的例子：

import torch
a = torch.tensor([[[1,2,3],[4,5,6]], [[7,8,9],[10,11,12]]])
print("a的shape:",a.shape)
b = torch.reshape(a,((4,3,1)))
print("b:",b)
print("b的shape:",b.shape)

输出结果为：
a的shape: torch.Size([2, 2, 3])
b: tensor([[[ 1],
[ 2],
[ 3]],

[[ 4],
[ 5],
[ 6]],

[[ 7],
[ 8],
[ 9]],

[[10],
[11],
[12]]])
b的shape: torch.Size([4, 3, 1])