pytorch transpose

PyTorch是一个开源的机器学习框架，它提供了丰富的工具和函数来支持深度学习任务。PyTorch中的transpose函数用于交换张量（Tensor）的维度顺序。

在PyTorch中，transpose函数的使用方式如下：

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

其中，input是要进行维度交换的张量，dim0和dim1是要交换的维度索引。

举个例子，假设有一个形状为(3, 4)的张量A，我们想要将其转置为形状为(4, 3)的张量B，可以使用transpose函数进行操作：

import torch

A = torch.randn(3, 4)
B = torch.transpose(A, 0, 1)

在上述代码中，我们将A的第0维和第1维进行了交换，得到了B。