torch.swapdims¶
- torch.swapdims(input, dim0, dim1) Tensor ¶
Alias for
torch.transpose()
.This function is equivalent to NumPy’s swapaxes function.
Examples:
>>> x = torch.tensor([[[0,1],[2,3]],[[4,5],[6,7]]]) >>> x tensor([[[0, 1], [2, 3]], [[4, 5], [6, 7]]]) >>> torch.swapdims(x, 0, 1) tensor([[[0, 1], [4, 5]], [[2, 3], [6, 7]]]) >>> torch.swapdims(x, 0, 2) tensor([[[0, 4], [2, 6]], [[1, 5], [3, 7]]])