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]]