mindspore.ops.ReverseV2

View Source On Gitee
class mindspore.ops.ReverseV2(axis)[source]

Reverses specific dimensions of a tensor.

Warning

The value range of “axis” is [-dims, dims - 1]. “dims” is the dimension length of “input_x”.

Parameters

axis (Union[tuple(int), list(int)]) – The indices of the dimensions to reverse.

Inputs:
  • input_x (Tensor) - The target tensor. The shape is \((N, *)\) where \(*\) means, any number of additional dimensions.

Outputs:

Tensor, has the same shape and type as input_x.

Raises
  • TypeError – If axis is neither list nor tuple.

  • TypeError – If element of axis is not an int.

  • ValueError – There are multiple identical axes in axis.

Supported Platforms:

Ascend GPU CPU

Examples

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