SoftXent To use SoftXent simply do: from soft_cross_entropy import SoftCrossEntropy xent = SoftCrossEntropy(mode) loss = xent(target, prediction)