比较与torch.Tensor.flip的差异

torch.Tensor.flip(dims)

mindspore.Tensor.flip(dims)

# PyTorch
import numpy as np
import torch
input = torch.tensor(np.arange(1, 9).reshape((2, 2, 2)))
output = input.flip(1)
print(output)
# tensor([[[3, 4],
#          [1, 2]],
#
#         [[7, 8],
#          [5, 6]]])

# MindSpore
import mindspore as ms
import mindspore.ops as ops

input = ms.Tensor(np.arange(1, 9).reshape((2, 2, 2)))
output = input.flip((1, ))
print(output)
# [[[3 4]
#   [1 2]]
#
#  [[7 8]
#   [5 6]]]