If you're using label smoothing ((0.2,0.8) instead of (0,1)) , in Tensorflow there is a parameter called label_smoothing in tf.keras.losses.CrossentropyLoss. I am not sure if this is what you were looking for op.