pytorch学习笔记1 - loss max

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

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

2. max


                
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值