LogSoftmax其实就是对softmax的结果进行log,即Log(Softmax(x)) n = nn.Softmax(dim=1) mn=n(input) m = mn.log() print(m) 等价于下面: m = nn.LogSoftmax(dim=1)
nn.LogSoftmax与Softmax区别
最新推荐文章于 2024-06-09 09:56:07 发布
LogSoftmax其实就是对softmax的结果进行log,即Log(Softmax(x)) n = nn.Softmax(dim=1) mn=n(input) m = mn.log() print(m) 等价于下面: m = nn.LogSoftmax(dim=1)