待办
返回对应给定中最大值的索引,方便进行和target结果的索引进行比较 索引方式见下 https://blog.csdn.net/liuweiyuxiang/article/details/84668269
pytorch之max()函数
原文:https://www.cnblogs.com/lishikai/p/12367626.html