log_loss
paddle.fluid.layers.
log_loss
(input, label, epsilon=0.0001, name=None)[源代码]
负log loss层
该 OP 对输入的预测结果和目标标签进行计算,返回负对数损失值。
- 参数:
- 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
代码示例
- import paddle.fluid as fluid
- label = fluid.layers.data(name='label', shape=[1], dtype='int64')
- prob = fluid.layers.data(name='prob', shape=[10], dtype='float32')
- cost = fluid.layers.log_loss(input=prob, label=label)