+-
我使用Keras和TensorFlow作为后端.如果我想对LSTM单元进行修改,例如“移除”输出门,我该怎么办？它是一个乘法门,所以不知怎的,我必须将它设置为固定值,以便无论倍增它,都没有效果.
最佳答案
首先,您应该定义您的 own custom layer.如果您需要一些直觉如何实现自己的单元格,请参阅Keras存储库中的 LSTMCell.例如.你的图层将是：
class MyLSTMCell(tf.keras.layers.Layer):
def build():
# define your own logic
def call():
# call your own logic
然后,使用tf.keras.layers.RNN来使用您的手机：
x = tf.keras.layers.RNN(my_custom_cell)(inputs)
点击查看更多相关文章
转载注明原文：python – 在Keras中定义自定义LSTM单元？ - 乐贴网
