pytorch的reshape用法

PyTorch的reshape函数可以将一个张量（tensor）重新排列为指定的形状。它接受一个整数或者一个元组作为参数，表示新的形状。例如，将一个形状为(2,3)的张量reshape为(3,2)的代码如下：

import torch

# 创建一个形状为(2,3)的张量
tensor = torch.tensor([[1,2,3],[4,5,6]])

# 使用reshape函数将其转换为(3,2)的形状
new_tensor = tensor.reshape(3,2)

print(new_tensor)

输出结果：

tensor([[1, 2],
        [3, 4],
        [5, 6]])