gongzb

摘要： 转自：https://zhuanlan.zhihu.com/p/525276061 首先说一下Softmax函数，公式如下: 1. 三维tensor(C,H,W) 一般会设置成dim=0,1,2,-1的情况(可理解为维度索引)。其中2与-1等价，相同效果。 用一张图片来更好理解这个参数dim数值变化 阅读全文