mindspore.ops.rot90

mindspore.ops.rot90(input, k, dims)[源代码]

沿轴指定的平面内将n-D Tensor旋转90度。 如果k>0,旋转方向是从第一轴朝向第二轴,如果k<0,旋转方向从第二轴朝向第一轴。

参数:
  • input (Tensor) - 所输入的Tensor。

  • k (int) - 旋转的次数。

  • dims (Union[list(int), tuple(int)]) - 要旋转的轴。

返回:

Tensor。

异常:
  • TypeError - 输入不是Tensor。

  • TypeError - k 不是整数。

  • TypeError - dims 不是整数组成的list或者tuple。

  • ValueError - dims 长度不为2。

  • ValueError - dims 中的元素不在输入Tensor的[-input.ndim, input.ndim)之间。

  • RuntimeError - dims 的两个元素相同。

支持平台:

Ascend GPU CPU

样例:

>>> import numpy as np
>>> from mindspore import Tensor, ops
>>> x = Tensor(np.array([[0, 1], [2, 3]])).astype(np.float32)
>>> k = 1
>>> dims = [0, 1]
>>> output = ops.rot90(x, k, dims)
>>> print(output)
[[1. 3.]
[0. 2.]]