pytorch的reshape函数

PyTorch中的reshape函数可以用来改变张量的形状，其语法为：

torch.reshape(input, shape)

其中，input是要改变形状的张量，shape是一个元组，表示改变后的形状。需要注意的是，新形状的元素个数必须与原形状相同。

例如，将一个形状为(2, 3, 4)的张量reshape成形状为(3, 8)的张量，可以使用以下代码：

import torch

x = torch.randn(2, 3, 4)
y = torch.reshape(x, (3, 8))
print(y.shape)

输出结果为：

torch.Size([3, 8])