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 CPU

样例：

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