import mindspore
from mindspore import mint, Tensor, ops, value_and_grad, context
import numpy as np

input_tensor = Tensor(np.array([[-0.1, 0.3, 3.6], [0.4, 0.5, -3.2]]), dtype = mindspore.float32)
index = Tensor(np.array([[0, 0], [1, 4]]), dtype = mindspore.int32)
output = mint.gather(input_tensor, 1, index)
print(output)

[[-0.1 -0.1]
 [ 0.5  0.5]]