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

torch.max(intup) 返回input矩阵的最大值 torch.max(intup,dim) 返回类型为元组，第一个元素为值，第二个元素为对应索引 dim = 0 按照列取最大值和索引并返回 dim = 1 按照行取最大值和索引并返回   squeeze() squeeze(input, dim=None)，如果不给定dim，则把input的所有size为1的维度给移除；如果给定dim，

>>阅读原文<<