对pytorch中softmax(dim=x)的理解

先看看文档怎么说的

这里说指定某个纬度（dim)，就会让那个纬度的和为1

看一下input是2维的情况

dim=0

dim=1

再看看input是3维的情况

dim = 0

dim = 1