torch.max()參數和用法解釋

torch.max(intup) 返回input矩陣的最大值 torch.max(intup,dim) 返回類型爲元組，第一個元素爲值，第二個元素爲對應索引 dim = 0 按照列取最大值和索引並返回 dim = 1 按照行取最大值和索引並返回   squeeze() squeeze(input, dim=None)，如果不給定dim，則把input的所有size爲1的維度給移除；如果給定dim，

>>阅读原文<<