smooth_l1

    该layer计算变量 x 和 的smooth L1 loss,它以 x 和 的第一维大小作为批处理大小。对于每个实例,按元素计算smooth L1 loss,然后计算所有loss。输出变量的形状是[batch_size, 1]

    参数:

    • inside_weight (Tensor|None) - 数据类型为float32,rank至少为2的张量。这个输入是可选的,与x的shape应该相同。如果给定, 的结果将乘以这个张量元素。

    • sigma (float|NoneType) - smooth L1 loss layer的超参数。标量,默认值为1.0。

    返回: smooth L1损失的输出值, shape为 [batch_size, 1]

    代码示例