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

最低0.47元/天 解锁文章
1347

被折叠的 条评论
为什么被折叠?



