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]))