torch transpose

torch.transpose函数用于交换张量的维度。

语法如下：

torch.transpose(input, dim0, dim1) → Tensor

参数说明：
- `input`：输入张量。
- `dim0`：第一个维度的索引。
- `dim1`：第二个维度的索引。

返回值：
返回一个新的张量，该张量为交换了指定维度的副本。

示例：

import torch

x = torch.tensor([[1, 2], [3, 4]])
y = torch.transpose(x, 0, 1)

print(x)
print(y)

输出结果：

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

在上面的示例中，我们将输入张量 `x` 的维度0和维度1进行了交换，得到了新的张量 `y`。