mindspore.numpy.rot90

查看源文件
mindspore.numpy.rot90(a, k=1, axes=(0, 1))[源代码]

将Tensor在指定的轴平面内旋转90度。旋转方向是从第一个轴指向第二个轴。

参数:
  • a (Tensor) - 输入的二维或更多维的Tensor。

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

  • axes (Union[tuple(int), list(int)],可选) - Tensor在由轴定义的平面内进行旋转。默认值: (0, 1) 。轴必须不同,并且shape为 (2,)

返回:

Tensor。

异常:
  • TypeError - 如果输入 a 不是Tensor,或参数 k 不是整数,或参数 axes 不是整数的tuple或list。

  • ValueError - 如果任何轴超出范围,或 axes 的长度不是2。

支持平台:

GPU

样例:

>>> import mindspore.numpy as np
>>> a = np.arange(24).reshape((2, 3, 4))
>>> output = np.rot90(a)
>>> print(output)
[[[ 8  9 10 11]
  [20 21 22 23]]
 [[ 4  5  6  7]
  [16 17 18 19]]
 [[ 0  1  2  3]
  [12 13 14 15]]]
>>> output = np.rot90(a, 3, (1, 2))
>>> print(output)
[[[ 8  4  0]
  [ 9  5  1]
  [10  6  2]
  [11  7  3]]
 [[20 16 12]
  [21 17 13]
  [22 18 14]
  [23 19 15]]]