center_loss

paddle.fluid.layers. center_loss ( input, label, num_classes, alpha, param_attr, update_center=True )

该 OP 接收一个来自于最后一个隐藏层的输出和目标标签作为输入，返回损失值。为每一个类别提供一个类别中心，计算 mini-batch 中每个样本与对应类别中心的距离的平均值作为 center loss。

对于输入，(X)和标签(Y)，计算公式为：

\[out = \frac{1}{2}(X - Y)^2\]

参数

  • input (Variable) - 输入形状为[N x M]的 2 维 Tensor，数据类型为 float32，float64。

  • label (Variable) - 输入的标签，一个形状为为[N x 1]的 2 维 Tensor，N 表示 batch size，数据类型为 int32。

  • num_class (int32) - 输入类别的数量。

  • alpha (float32|float64|Variable) - 学习率。数据类型为 float32 或者 float64。

  • param_attr (ParamAttr) - 指定权重参数属性的对象。具体用法请参见 ParamAttr 。

  • update_center (bool) - 是否更新类别中心的参数。

返回

形状为[N x 1]的 2 维 Tensor。

返回类型

Variable

代码示例

COPY-FROM: paddle.fluid.layers.center_loss