Function Differences with torch.amin

The following mapping relationships can be found in this file.

torch.amin

torch.amin(input, dim, keepdim=False, *, out=None) -> Tensor

For more information, see torch.amin.

mindspore.ops.amin

mindspore.ops.amin(x, axis=(), keepdims=False) -> Tensor

For more information, see mindspore.ops.amin.

Differences

PyTorch: Find the minimum element of input according to the specified dim. keepdim controls whether the output and the input have the same dimension. out can get the output.

MindSpore: Find the minimum element of x according to the specified axis. The keepdims function is identical to PyTorch. MindSpore does not have out parameter. MindSpore axis has a default value, and finds the minimum value of all elements of x if axis is the default value.

Code Example

# PyTorch
import torch

input = torch.tensor([[1, 2, 3], [3, 2, 1]], dtype=torch.float32)
print(torch.amin(input, dim=0, keepdim=True))
# tensor([[1., 2., 1.]])

# MindSpore
import mindspore

x = mindspore.Tensor([[1, 2, 3], [3, 2, 1]], dtype=mindspore.float32)
print(mindspore.ops.amin(x, axis=0, keepdims=True))
# [[1. 2. 1.]]