Keras双向RNN-初始化不适用于tf.zeros()

时间:2019-06-06 19:16:26

标签: tensorflow keras deep-learning recurrent-neural-network

我在tensorflow 1.12 / 13/14的渴望模式下工作。

这是初始化:

gru = keras.layers.GRU(units, return_sequences=True, return_state=True,
                       recurrent_activation='sigmoid',
                       recurrent_initializer='glorot_uniform')
bigru = keras.layers.Bidirectional(gru)  

这有效:

bigru(x, initial_state=bigru(x)[1:])

这不是:

bigru(x, initial_state=[tf.zeros((16,32)), tf.zeros((16,32))])

这很奇怪,因为bigru(x)[1:]返回[EagerTensor, EagerTensor]的列表。这与zeros列表的类型和大小相同。

这是一个错误吗?有人知道如何解决吗?

我将非常感谢您的帮助!

0 个答案:

没有答案