softmax避坑，训练集80、90准确率，测试集一直50左右

在使用softmax之后，大家一定要把outputs输出查看是否满足每一行或者每一列值相加为1（看自己输出是一列还是一行）

（1）dim=0：对每一列的所有元素进行softmax运算，并使得每一列所有元素和为1。

（2）dim=1：对每一行的所有元素进行softmax运算，并使得每一行所有元素和为1。