def METHOD_NAME(self, prediction, label):
    mask_lm_loss, lm_softmax = nn.functional.softmax_with_cross_entropy(
        logits=prediction, label=label, return_softmax=True)
    mean_mask_lm_loss = paddle.mean(mask_lm_loss)
    return mean_mask_lm_loss