mindspore.mint.flip

mindspore.mint.flip(input, dims)[源代码]

沿指定维度翻转tensor中元素。

参数：

input (Tensor) - 输入tensor。
dims (Union[list[int], tuple[int]]) - 翻转的维度。

返回：

Tensor

支持平台：

Ascend

样例：

>>> import mindspore
>>> input = mindspore.tensor(mindspore.ops.arange(1, 9).reshape((2, 2, 2)))
>>> output = mindspore.mint.flip(input, (0, 2))
>>> print(output)
[[[6 5]
  [8 7]]
 [[2 1]
  [4 3]]]