tf.contrib.keras.backend.in_train_phase

tf.contrib.keras.backend.in_train_phase

in_train_phase(
    x,
    alt,
    training=None
)

Defined in tensorflow/contrib/keras/python/keras/backend.py.

Selects x in train phase, and alt otherwise.

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)
    specifing the learning phase.

Returns:

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