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

【算子功能描述】

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

該算子的計算公式爲：output = sum(t**2)/2  即：對輸入x中的每個值求平方，而後所有相加求和，再將和除以2，輸出是一個標量。

【模型結構】

 

【算子定義】

該算子的定義在 tensorflow/core/ops/nn_ops.cc ，具體定義以下：

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