.argmax(-1)理解

返回最后一个维度max之后的索引数组

https://blog.csdn.net/sinat_29047129/article/details/103661475?depth_1-utm_source=distribute.pc_relevant.none-task&utm_source=distribute.pc_relevant.none-task