binary_cross_entropy_with_logits

F.binary_cross_entropy_with_logits(input, target, weight=None, size_average=None, reduce=None, reduction='mean', pos_weight=None)

参数

  • weight:各类别的loss设置权值
  • ignore_index:忽略某个类别
  • reduction:计算模式,可为none/sum/mean
    • none:逐个元素计算
    • sum:所有元素求和,返回标量
    • mean:加权平均,返回标量

你可能感兴趣的:(深度学习,python,pytorch)