mindspore.ops.reverse

查看源文件
mindspore.ops.reverse(x, axis)[源代码]

对输入Tensor按指定维度反转。

警告

“axis”的取值范围为[-dims, dims - 1],”dims”表示”x”的维度长度。

参数:
  • x (Tensor) - 输入Tensor。其shape为: \((N, *)\) ,其中 \(*\) 表示任意数量的附加维度。

  • axis (Union[tuple(int), list(int)]) - 指定反转的轴。

输出:

Tensor,shape和数据类型与输入 x 相同。

异常:
  • TypeError - axis 既不是list也不是tuple。

  • TypeError - axis 的元素不是int。

支持平台:

Ascend GPU CPU

样例:

>>> import mindspore
>>> import numpy as np
>>> from mindspore import Tensor, ops
>>> input_x = Tensor(np.array([[1, 2, 3, 4], [5, 6, 7, 8]]), mindspore.int32)
>>> output = ops.reverse(input_x, axis=[1])
>>> print(output)
[[4 3 2 1]
 [8 7 6 5]]
>>> input_x = Tensor(np.array([[1, 2, 3, 4], [5, 6, 7, 8]]), mindspore.int32)
>>> output = ops.reverse(input_x, axis=[1, 0])
>>> print(output)
[[8 7 6 5]
 [4 3 2 1]]