torch.min() 和np.min()

 torch.min(tensor, dim) 的返回值分别为指定维度的最小值和对应的索引

y_score = torch.tensor([[0.9, 0.05, 0.05],
                    [0.1, 0.8, 0.1],
                    [0.05, 0.1, 0.85],
                    [0.8, 0.15, 0.05],
                    [0.2, 0.7, 0.1],
                    [0.1, 0.25, 0.65]])
y_min = torch.min(y_score, dim=1)

y_min
torch.return_types.min(
values=tensor([0.0500, 0.1000, 0.0500, 0.0500, 0.1000, 0.1000]),
indices=tensor([1, 0, 0, 2, 2, 0]))

np.min(array, axis) 返回值只有一个，即指定维度最小值组成的array

y_score = np.array([[0.9, 0.05, 0.05],
                    [0.1, 0.8, 0.1],
                    [0.05, 0.1, 0.85],
                    [0.8, 0.15, 0.05],
                    [0.2, 0.7, 0.1],
                    [0.1, 0.25, 0.65]])
y_min = np.min(y_score, axis=1)

y_min
array([0.05, 0.1 , 0.05, 0.05, 0.1 , 0.1 ])