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’:计算输出元素的总和。
11-01
4466