softmax在代码的哪里？

Question

卷积神经网络代码中softmax在哪里啊？代码中没找到，求问在代码的哪个地方（pytorch）

Answer 1

• 帮你找了个相似的问题, 你可以看下: https://ask.csdn.net/questions/7649031
• 这篇博客也不错, 你可以看下softmax pytorch从零实现的代码
• 除此之外, 这篇博客: 手把手教你利用PyTorch实现图像识别中的 3.3 softmax运算和交叉熵损失函数 部分也许能够解决你的问题, 你可以仔细阅读以下内容或跳转源博客中阅读:
  

• 分开定义softmax运算和交叉熵损失函数会造成数值不稳定。因此PyTorch提供了一个具有良好数值稳定性且包括softmax运算和交叉熵计算的函数。

  loss = nn.CrossEntropyLoss()
  

• 您还可以看一下 唐宇迪老师的深度学习框架-PyTorch实战课程中的 特征分配与softmax机制小节, 巩固相关知识点