binary_cross_entropy_with_logits 该函数参数:
- logits (Tensor) - 输入预测值。其数据类型为float16或float32。
- label (Tensor) - 输入目标值,shape与 logits 相同。数据类型为float16或float32。
- weight (Tensor,可选) - 指定每个批次二值交叉熵的权重。支持广播,使其shape与 logits 的shape保持一致。数据类型必须为float16或float32。默认值:None , weight 是值为 1 的Tensor。
- pos_weight (Tensor,可选) - 指定正类的权重。是一个长度等于分类数的向量。支持广播,使其shape与 logits 的shape保持一致。数据类型必须为float16或float32。默认值:None , pos_weight 是值为 1 的Tensor。
- reduction (str,可选) - 指定应用于输出结果的规约计算方式,可选 ‘none’ 、 ‘mean’ 、 ‘sum’ ,默认值: ‘mean’ 。- ‘none’:不应用规约方法。- ‘mean’:计算输出元素的加权平均值。- ‘sum’:计算输出元素的总和。
本文转载自: https://blog.csdn.net/qq_75125305/article/details/140079722
版权归原作者 AbsoluteClownMaster 所有, 如有侵权,请联系我们删除。
版权归原作者 AbsoluteClownMaster 所有, 如有侵权,请联系我们删除。