比较与torch.min的功能差异
torch.min
torch.min(
input,
dim,
keepdim=False,
out=None
)
更多内容详见torch.min。
mindspore.ops.ArgMinWithValue
class mindspore.ops.ArgMinWithValue(
axis=0,
keep_dims=False
)(input_x)
使用方式
PyTorch:输出为元组(最小值, 最小值的索引)。
MindSpore:输出为元组(最小值的索引, 最小值)。
代码示例
import mindspore as ms
import mindspore.ops as ops
import torch
import numpy as np
# Output tuple(index of min, min).
input_x = ms.Tensor(np.array([0.0, 0.4, 0.6, 0.7, 0.1]), ms.float32)
argmin = ops.ArgMinWithValue()
index, output = argmin(input_x)
print(index)
print(output)
# Out:
# 0
# 0.0
# Output tuple(min, index of min).
input_x = torch.tensor([0.0, 0.4, 0.6, 0.7, 0.1])
output, index = torch.min(input_x, 0)
print(index)
print(output)
# Out:
# tensor(0)
# tensor(0.)