【d2l】【常见函数】【14】 nn.CrossEntropyloss 和 d2l.cross_entropy
** nn.CrossEntropyloss自带softmax功能,d2l.cross_entrop不带softmax功能**
Note reduction = 'none'
nn.CrossEntropyloss的常见用法:
loss = nn.CrossEntropyloss(reduction='none')
l = loss(input_y_hat, y_label)
那么这个reducion是干什么用的呢?
参考 https://pytorch.org/docs/stable/generated/torch.nn.CrossEntropyLoss.html
选了 reduction = 'none'
的话就不会对结果求和或求均值