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]
代码示例