pytorch学习笔记1 - loss max

最新推荐文章于 2023-07-24 09:09:51 发布

原创

最新推荐文章于 2023-07-24 09:09:51 发布 · 992 阅读

0 ·

CC 4.0 BY-SA版权

这篇博客介绍了PyTorch中的loss函数，包括其参数如weight、size_average、ignore_index和reduce的使用。同时，文章还详细讲解了如何进行max操作，指出max函数可以返回最大值及其对应的索引，可以根据第二个参数按列或行求最大值。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1. loss函数

https://pytorch.org/docs/stable/nn.html#loss-functions

torch.nn.CrossEntropyLoss(weight=None, size_average=True, ignore_index=-100, reduce=True)

参数：
weight (Tensor, optional) – a manual rescaling weight given to each class. If given, has to be a Tensor of size C
size_average (bool, optional) : False - sum of mibi-batch; True - average of mibi-batch.
ignore_index (int, optional) – 是否忽略某特定index对gradient的贡献，size_average = True, 忽略指定index目标的loss.
reduce (bool, optional) – False - 返回单个样本的loss， True - 按照size_average设定来计算loss