torch.max(input, dim, keepdim=False, out=None) -> (Tensor, LongTensor)
max函数需要注意的是,它是一个过载函数,函数参数不同函数的功能和返回值也不同。
input是一个tensor.
当max函数中有维数参数的时候,它的返回值为两个,一个为最大值,另一个为最大值的索引.
对于dim参数,以一个二维数据输入为例,当dim为0时,max()返回的是每列的最大值以及各列最大值的索引;当dim为1时,max()返回的是每行的最大值以及各行最大值的索引。
a = t.randn(3, 4)
print(a)
print('维度为0:')
print(t.max(a, 0))
print('维度为1:')
print(t.max(a, 1))
输出为:
tensor([[-1.2851, 1.2206, -0.2578, 1.4101],
[ 0.6201, -0.8447, 0.0455, -1.0944],
[ 0.8037, -1.2278, -0.3627, 0.0303]])
维度为0:
torch.return_types.max(
values=tensor([0.8037, 1.2206, 0.0455, 1.4101]),
indices=tensor([2, 0, 1, 0]))
维度为1:
torch.return_types.max(
values=tensor([1.4101, 0.6201, 0.8037]),
indices=tensor([3, 0, 0]))