torch.nn.CrossEntropyLoss() 参数、计算过程以及及输入Tensor ...2022-08-24
1.参数 torch.nn.CrossEntropyLoss ( weight=None, size_average=None, ignore_index=-100, reduce=None, reduction='mean', label_smoothing=0.0) 最常用的参数为 reduction ( str, optional) ,可设置其值为 mean, sum, none ,默认为 mean 。 该参数主要影响多个样本输入时,损失的综合方法。 mean表示损失为多个样本的平均值,sum表示损失的和,none
了解详情