本文介绍如何重用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