本文介紹如何重用tf.keras.layers
中的層。
在tensorflow中,我們可以通過variable_scope
和reuse
參數控制變量的初始化和重用。但當我們在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
的關鍵在於:一次定義,多次調用