python中的max()与min()的用法介绍

A是一个 m 行 n 列的矩阵：

A.min(0) : 返回A每一列最小值组成的一维数组；
A.min(1)：返回A每一行最小值组成的一维数组；
A.max(0)：返回A每一列最大值组成的一维数组；
A.max(1)：返回A每一行最大值组成的一维数组；
这里的0表示行，1表示列

我利用tensor写的张量，里面还可以显示每一行每一列输出的最大最小的索引

A = torch.rand(3,4)
print(A)
print(A.min(0))
print(A.min(1))
print(A.max(0))
print(A.max(1))

输出结果：

tensor([[0.8873, 0.5165, 0.1780, 0.5059],
        [0.4438, 0.2884, 0.2477, 0.4414],
        [0.9050, 0.1961, 0.1676, 0.7914]])
torch.return_types.min(
values=tensor([0.4438, 0.1961, 0.1676, 0.4414]),
indices=tensor([1, 2, 2, 1]))
torch.return_types.min(
values=tensor([0.1780, 0.2477, 0.1676]),
indices=tensor([2, 2, 2]))
torch.return_types.max(
values=tensor([0.9050, 0.5165, 0.2477, 0.7914]),
indices=tensor([2, 0, 1, 2]))
torch.return_types.max(
values=tensor([0.8873, 0.4438, 0.9050]),
indices=tensor([0, 0, 0]))