.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