torch.max()参数和用法解释

torch.max(input)

返回input矩阵的最大值

torch.max(input,dim)

返回类型为元组(namedtuple)，第一个元素为值，第二个元素为对应索引

dim = 0 按照列取最大值和索引并返回

dim = 1 按照行取最大值和索引并返回

 

squeeze()

squeeze(input, dim=None)，如果不给定dim，则把input的所有size为1的维度给移除；如果给定dim，则只移除给定的且size为1的维度。