tensor 如何转置

在 PyTorch 中，可以使用 `torch.transpose()` 函数来实现 tensor 的转置操作。该函数的语法如下：

torch.transpose(input, dim0, dim1) -> Tensor

其中，`input` 表示待转置的 tensor，`dim0` 和 `dim1` 分别表示要交换的维度。例如，对于一个 2 维的 tensor，可以通过以下代码实现转置：

import torch

# 定义一个 2x3 的 tensor
x = torch.tensor([[1, 2, 3], [4, 5, 6]])

# 转置 x，将第一维和第二维交换
y = torch.transpose(x, 0, 1)

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

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

上述代码中，`x` 是一个 2x3 的 tensor，通过 `torch.transpose(x, 0, 1)` 实现了转置，将第一维和第二维交换，得到了一个 3x2 的 tensor `y`。