reuse tf.keras.layers

本文介绍如何重用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的关键在于：一次定义，多次调用