阅读(5.6k) 书签 (0)

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,要么返回alttraining标志默认为K.learning_phase()