mindspore.mint.flip

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

沿给定轴翻转Tensor中元素的顺序。

Tensor的shape会被保留,但是元素将重新排序。

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

  • dims (Union[list[int], tuple[int]]) - 需要翻转的一个轴或多个轴。在元组中指定的所有轴上执行翻转,如果 dims 是一个包含负数的整数元组,则该轴为按倒序计数的轴位置。

返回:

返回沿给定轴翻转计算结果的Tensor。

异常:
  • TypeError - input 不是Tensor。

  • ValueError - dims 为None。

  • ValueError - dims 不为int组成的list或tuple。

支持平台:

Ascend

样例:

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