mindspore.ops.ReverseV2
- class mindspore.ops.ReverseV2(axis)[源代码]
对输入Tensor按指定维度反转。
Warning
“axis”的取值范围为[-dims, dims - 1],”dims”表示”input_x”的维度长度。
参数:
axis (Union[tuple(int), list(int)]) - 指定反转的轴。
输入:
input_x (Tensor) - 输入需反转的任意维度的Tensor。数据类型为数值型,不包括float64。shape: \((N, *)\) ,其中 \(*\) 表示任意数量的附加维度。
输出:
Tensor,shape和数据类型与输入 input_x 相同。
异常:
TypeError - axis 既不是list也不是tuple。
TypeError - axis 的元素不是int。
- 支持平台:
Ascend
GPU
样例:
>>> input_x = Tensor(np.array([[1, 2, 3, 4], [5, 6, 7, 8]]), mindspore.int32) >>> op = ops.ReverseV2(axis=[1]) >>> output = op(input_x) >>> print(output) [[4 3 2 1] [8 7 6 5]] >>> op = ops.ReverseV2(axis=[1, 0]) >>> output = op(input_x) >>> print(output) [[8 7 6 5] [4 3 2 1]]