包装器 tf.keras.layers.Bidirectional() 介绍

657 阅读1分钟

功能
实现RNN双向构造,比方说LSTM、GRU.其中Bidirectional的意思为双向的
参数

tf.keras.layers.Bidirectional(
    layer, merge_mode='concat', weights=None, backward_layer=None,
    **kwargs
)
  • layer:选择模型.如LSTM、GRU.
  • merge_mode:前向和后向RNN的输出将被组合的模式。{‘sum’,‘mul’,‘concat’,‘ave’,None}中的一个。如果为None,则将不合并输出,它们将作为列表返回。默认值为“ concat”.
  • weights:#
  • backward_layer:处理后向输入处理的神经网络,如果未提供,则作为参数传递的图层实例 将用于自动生成后向图层.

例:\

x = Bidirectional(LSTM(128,return_sequences=True))(embedding)