pytorch torch.transpose

交换维度顺序

应用

>>> x = torch.randn(2, 3)
>>> x
tensor([[ 1.0028, -0.9893,  0.5809],
        [-0.1669,  0.7299,  0.4942]])
>>> torch.transpose(x, 0, 1)
tensor([[ 1.0028, -0.1669],
        [-0.9893,  0.7299],
        [ 0.5809,  0.4942]])

API

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

参数	描述
input (Tensor)	the input tensor.
dim0 (int)	the first dimension to be transposed
dim1 (int)	the second dimension to be transposed

参考：
https://pytorch.org/docs/stable/generated/torch.transpose.html#torch.transpose