参考了下面的两个链接: softmax: https://blog.youkuaiyun.com/q2519008/article/details/107086024?utm_medium=distribute.pc_relevant_t0.none-task-blog-BlogCommendFromMachineLearnPai2-1.channel_param&depth_1-utm_source=distribute.pc_relevant_t0.none-task-blog-BlogCommendFromMachineLearnPai2-1.channel_param 其中如果axis的参数为-1,则表示最后一个维度 argmax: 看下面的例子: