0


binary_cross_entropy_with_logits函数的参数设定

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 所有, 如有侵权,请联系我们删除。

“binary_cross_entropy_with_logits函数的参数设定”的评论:

还没有评论