tf.keras.constraints.UnitNorm