mindspore.Tensor.rot90

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

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

参数:
  • k (int) - 旋转的次数。默认值:1。

  • dims (Union[list(int), tuple(int)]) - 要旋转的轴。默认值:[0,1]。

返回:

Tensor。

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

  • TypeError - k 不是整数。

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

  • ValueError - dims 长度不为2。

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

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

支持平台:

Ascend GPU

样例:

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