tf.keras.backend.one_hot