如何写我自己的Keras图层?

时间:2018-04-15 16:09:30

标签: python tensorflow keras rnn

我不清楚如何编写我自己的Keras图层,如Keras Docs所示。 我想要做的是使用下面的代码来定义Keras层,但我不知道如何。谁能给我一个帮助?非常感谢。

first_layer = MyRNNCell(n_neurons,)
second_layer = MyRNNCell(n_neurons)
cell = tf.nn.rnn_cell.MultiRNNCell([first_layer, second_layer])
output, state = tf.nn.dynamic_rnn(cell, X, dtype=tf.float32)

其中MyRNNCell是我自己定义的函数。

0 个答案:

没有答案