sigmoid_cross_entropy_with_logits
sigmoid_cross_entropy_with_logits
(x, label, ignore_index=-100, name=None, normalize=False)[源代码]
logistic loss可通过下式计算:
已知:
代入上方计算logistic loss公式中:
为了计算稳定性,防止
输入 和 label
都可以携带LoD信息。然而输出仅采用输入 的LoD。
- x (Variable) - (Tensor, 默认 Tensor
),形为 N x D 的二维张量,N为batch大小,D为类别数目。该输入是一个由先前运算得出的logit组成的张量。logit是未标准化(unscaled)的log概率, 公式为 , 数据类型为float32或float64。
- ignore_index (int) - (int,默认kIgnoreIndex)指定被忽略的目标值,它不会影响输入梯度。
- name (str,可选) – 具体用法请参见 ,一般无需设置,默认值为None。
- x (Variable) - (Tensor, 默认 Tensor
返回: Variable(Tensor, 默认Tensor<float>), 形为 N x D 的二维张量,其值代表了按元素的logistic loss,数据类型为float32或float64。
代码示例