tf.keras.backend.in_train_phase

View source on GitHub

Selects x in train phase, and alt otherwise.

View aliases

Compat aliases for migration

See Migration guide for more details.

tf.compat.v1.keras.backend.in_train_phase, `tf.compat.v2.keras.backend.in_train_phase`

tf.keras.backend.in_train_phase(
    x, alt, training=None
)

Note that alt should have the same shape as x.

Arguments
x	What to return in train phase (tensor or callable that returns a tensor).
alt	What to return otherwise (tensor or callable that returns a tensor).
training	Optional scalar tensor (or Python boolean, or Python integer) specifying the learning phase.

Returns
Either x or alt based on the training flag. the training flag defaults to K.learning_phase().