【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
image

选了 reduction = 'none' 的话就不会对结果求和或求均值

posted @ 2023-07-28 11:16  zz子木zz  阅读(26)  评论(0编辑  收藏  举报