您可以通过设置在LSTM层中启用状态性stateful=True
。这将改变该层的行为,使其始终使用该层的上一次调用的状态,而不是为每个层重置它的状态layer.call(x)
。
例如,具有32个单位的LSTM层,批处理大小为1,序列长度为64,特征长度为10:
LSTM(32, stateful=True, batch_input_shape=(1,64,10))
通过此连续调用,predict
将使用以前的状态。
本文收集自互联网,转载请注明来源。
如有侵权,请联系[email protected] 删除。
我来说两句