我不清楚如何编写我自己的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是我自己定义的函数。