pytorch reshape

在 PyTorch 中，可以使用 `reshape` 方法来改变张量的形状。`reshape` 方法接受一个元组作为参数，指定新的形状。下面是一个示例：

import torch

x = torch.tensor([[1, 2, 3],
                  [4, 5, 6]])

# 使用 reshape 方法改变形状
y = x.reshape((3, 2))

print(y)

输出：

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

在上述示例中，我们将一个形状为 (2, 3) 的张量 `x` 转换成了一个形状为 (3, 2) 的张量 `y`。`reshape` 方法会按照元素在内存中的顺序来重新排列张量的形状。请注意，`reshape` 方法返回的是一个新的张量，原始张量 `x` 的形状不会改变。