torch.rot90¶
- torch.rot90(input, k=1, dims=(0, 1)) Tensor ¶
在由 dims 轴指定的平面内将 n 维张量旋转 90 度。如果 k > 0,则旋转方向是从第一个轴到第二个轴,如果 k < 0,则从第二个轴到第一个轴。
示例
>>> x = torch.arange(4).view(2, 2) >>> x tensor([[0, 1], [2, 3]]) >>> torch.rot90(x, 1, [0, 1]) tensor([[1, 3], [0, 2]]) >>> x = torch.arange(8).view(2, 2, 2) >>> x tensor([[[0, 1], [2, 3]], [[4, 5], [6, 7]]]) >>> torch.rot90(x, 1, [1, 2]) tensor([[[1, 3], [0, 2]], [[5, 7], [4, 6]]])