卷积神经网络代码中softmax在哪里啊?代码中没找到,求问在代码的哪个地方(pytorch)
分开定义softmax运算和交叉熵损失函数会造成数值不稳定。因此PyTorch提供了一个具有良好数值稳定性且包括softmax运算和交叉熵计算的函数。
loss = nn.CrossEntropyLoss()
softmax一般是最后一层用来做为激活层的,你好好找找贴出代码。如果找不到,也不排除根本没有。一般来说分类问题才用softmax