结论： torch.max(),不指定dim参数，返回输入张量中所有数据的最大值； 如果指定dim参数，则返回按照指定维度的最大值和各最大值对应的位置 torch.argmax(),不指定dim，然后输入张量中所有数据最大值位置（按照张量被拉伸为1为向量算）； 如果指定dim，返回指定维度的最大值位置，另外可以通过keepdim来保留原张量的形状
torch.max和torch.argmax
原文：https://www.cnblogs.com/pyclq/p/15071494.html
