torch 数据维度更换

要更改 PyTorch 张量的维度，可以使用 `torch.view()` 函数。这个函数允许你重新定义张量的形状，只要新形状的元素数量与原始形状相同。

下面是一个示例，展示如何更改张量的维度：

import torch

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

# 使用 view 函数将其转换为 3x2 的张量
y = x.view(3, 2)

print(y)

输出:

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

此示例中，我们创建了一个 2x3 的张量 `x`，然后使用 `view` 函数将其转换为 3x2 的张量 `y`。