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