pytorch常用函数

1.torch.max(input, dim, keepdim=False, *, out=None)

返回一个命名元组，其中是给定维度中张量 每一行的最大值。并且是找到的每个最大值的索引位置 (argmax)。(values, indices)valuesinputdimindices

如果keepdim为True，则输出张量与输入张量的大小相同，但维度dim中的输出张量大小为1。否则，dim会被压缩（请参见torch.squege（）），导致输出张量的维度比输入张量少1。