当前位置:   article > 正文

BCELoss和BCEWithLogitsLoss

bcewithlogitsloss

1. BCELoss

该类主要用来创建衡量目标和输出之间的二进制交叉熵的标准。
用法如下:
torch.nn.BCELoss(weight=None, size_average=None, reduce=None, reduction='mean')

参数:

  • weight,表示对loss中每个元素的加权权值;
  • reduction, 指定输出的格式,包括'none','mean','sum';
  • 其它两个参数一般不推荐使用;

形状:

  • input,(N, *);
  • target,(N, *);
  • output,标量,当reduction为'none'时,为(N, *)。

计算

当reduction参数为'none',即对计算结果不进行处理时,loss可以表示为,

其中,

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/weixin_40725706/article/detail/708938
推荐阅读
相关标签
  

闽ICP备14008679号