argmax

得到最大值的索引号

dim=0表示行维度不要了

dim=1表示列维度不要了

【推广】 免费学中医，健康全家人
原文地址：https://www.cnblogs.com/liujianing/p/13122033.html