mindspore.ops.flip

查看源文件
mindspore.ops.flip(input, dims)[源代码]

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

参数:
  • input (Tensor) - 输入tensor。

  • dims (Union[list[int], tuple[int]]) - 翻转的维度。

返回:

Tensor

支持平台:

Ascend GPU CPU

样例:

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