Rate this Page
★★★★★

torch.rot90#

torch.rot90(input,k=1,dims=(0,1))→Tensor#

Rotate an n-D tensor by 90 degrees in the plane specified by dims axis.Rotation direction is from the first towards the second axis if k > 0, and from the second towards the first for k < 0.

Parameters

  • input (Tensor) – the input tensor.

  • k (int) – number of times to rotate. Default value is 1

  • dims (a list ortuple) – axis to rotate. Default value is [0, 1]

Example:

>>>x=torch.arange(4).view(2,2)>>>xtensor([[0, 1],        [2, 3]])>>>torch.rot90(x,1,[0,1])tensor([[1, 3],        [0, 2]])>>>x=torch.arange(8).view(2,2,2)>>>xtensor([[[0, 1],         [2, 3]],        [[4, 5],         [6, 7]]])>>>torch.rot90(x,1,[1,2])tensor([[[1, 3],         [0, 2]],        [[5, 7],         [4, 6]]])