TensorFlow函数教程:tf.keras.backend.in_train_phase
2019-03-13 17:35 更新
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()
。