center_loss

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

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

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

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

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

    返回：形状为[N x 1]的2维Tensor|LoDTensor。

    代码示例：