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

以上内容是否对您有帮助:
在线笔记
App下载
App下载

扫描二维码

下载编程狮App

公众号
微信公众号

编程狮公众号

意见反馈
返回顶部