argmax

得到最大值的索引号

dim=0表示行维度不要了

dim=1表示列维度不要了

posted @ 2020-06-13 22:27  6+0  阅读(165)  评论(0)  编辑  收藏  举报