【TensorFlow】【算子解析】【tf.nn】tf.nn.l2_loss

时间 2019-11-15

标签 TensorFlow 算子解析 tf.nn tf.nn.l2 loss 繁體版

原文原文链接

【算子功能描述】code

tf.nn.l2_loss(
    t,
    name=None
)

该算子的计算公式为：output = sum(t**2)/2 即：对输入x中的每个值求平方，而后所有相加求和，再将和除以2，输出是一个标量。blog

【模型结构】tensorflow

【算子定义】float

该算子的定义在 tensorflow/core/ops/nn_ops.cc ，具体定义以下：im

REGISTER_OP("L2Loss")
    .Input("t: T")
    .Output("output: T")
    .Attr("T: {half, bfloat16, float, double}")
    .SetShapeFn(shape_inference::ScalarShape);