log_loss

  • paddle.fluid.layers.log_loss(input, label, epsilon=0.0001, name=None)[源代码]

负log loss层

该 OP 对输入的预测结果和目标标签进行计算,返回负对数损失值。

log_loss - 图1

  • 参数:
    • input (Variable) – 形为 [N x 1] 的二维张量, 其中 N 为 batch 大小。该输入是由前驱算子计算得来的概率, 数据类型是 float32。
    • label (Variable) – 形为 [N x 1] 的二维张量,真值标签, 其中 N 为 batch 大小,数据类型是 float32。
    • epsilon (float) – epsilon
    • name (str,可选) – 具体用法请参见 Name ,一般无需设置,默认值为None。

返回: 形为[N x 1]的二维张量,计算出的负log_loss值,数据类型为 float32

返回类型: Variable

代码示例

  1. import paddle.fluid as fluid
  2. label = fluid.layers.data(name='label', shape=[1], dtype='int64')
  3. prob = fluid.layers.data(name='prob', shape=[10], dtype='float32')
  4. cost = fluid.layers.log_loss(input=prob, label=label)