TensorFlow函數(shù)教程:tf.keras.backend.in_train_phase

2019-03-13 17:35 更新

tf.keras.backend.in_train_phase函數(shù)

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

定義在:tensorflow/python/keras/backend.py。

在訓練階段選擇x,其他階段選擇alt。

請注意alt應該與x的shape相同。

參數(shù):

  • x:在訓練階段需要返回的x (張量或返回張量的可調(diào)用函數(shù))。
  • alt:在其他階段需要返回的 alt (張量或返回張量的可調(diào)用函數(shù))。
  • training:可選的標量張量(或Python布爾值,或Python整數(shù)),指定學習階段。

返回:

基于training標志,要么返回x,要么返回alt。training標志默認為K.learning_phase()

以上內(nèi)容是否對您有幫助:
在線筆記
App下載
App下載

掃描二維碼

下載編程獅App

公眾號
微信公眾號

編程獅公眾號