TensorFlow函数教程：tf.keras.backend.in_train_phase

tf.keras.backend.in_train_phase函数

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

定义在：tensorflow/python/keras/backend.py。

在训练阶段选择x，其他阶段选择alt。

请注意alt应该与x的shape相同。

参数：

• x：在训练阶段需要返回的x (张量或返回张量的可调用函数)。
• alt：在其他阶段需要返回的 alt (张量或返回张量的可调用函数)。
• training：可选的标量张量（或Python布尔值，或Python整数），指定学习阶段。

返回：

基于training标志，要么返回x，要么返回alt。training标志默认为K.learning_phase()。