reuse tf.keras.layers

本文介紹如何重用tf.keras.layers中的層。

在tensorflow中,我們可以通過variable_scopereuse參數控制變量的初始化和重用。但當我們在tensorflow中使用tf.keras.layers時,這種參數重用的方式就失效了。

不幸的是,tensorflow文檔不全,而且總是誘導我們升級版本,扶正keras。那麼,應該如何在tensorflow的代碼中使用tf.keras.layers,並且能夠重用其中的參數呢?

重用keras.layers中的參數

# 定義layer
rnn = tf.keras.layers.LSTM(64, activation='tanh')
# 第一次調用
out = rnn(x)
# 重用
out_reuse = rnn(x_reuse)

此時,x=>out中的參數和x_reuse=>out_reuse中的參數是一致的。

重用tf.keras.layers的關鍵在於:一次定義,多次調用

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章