def METHOD_NAME(config, predictions):
    """doc
    """
    logits = predictions["logits"]
    labels = paddle.unsqueeze(
        paddle.arange(
            0, paddle.shape(logits)[0], dtype="int64"), axis=1)
    loss = paddle.nn.functional.softmax_with_cross_entropy(logits, labels)
    loss = paddle.mean(loss)
    return loss