You need to enable JavaScript to run this app.
最新活动
大模型
产品
解决方案
定价
生态与合作
支持与服务
开发者
了解我们

如何构建多层双向循环神经网络(RNN)?附实现尝试代码

BiLSTM 实现示例(Type1)

我在网上翻了好几篇技术文章,试了好几种实现方案,这里给你分享其中一种基于TensorFlow的BiLSTM实现方式:

def biLSTM(inputs, n_layers, seq_len, n_hidden, batch_size):
    output = inputs
    for n in range(n_layers):
        lstm_fw = tf.nn.rnn_cell.LSTMCell(n_hidden, state_is_tuple=True)
        lstm_bw = tf.nn.rnn_cell.LSTMCell(n_hidden, state_is_tuple=True)
        _initial_state_fw = lstm_fw.zero_state(batch_size, tf.float32)
        _initial_state_bw = lstm_bw.zero_state(batch_size, tf.float32)
        output, _states = tf.nn.bidirectional_dynamic_rnn(lstm_fw, ...)

内容的提问来源于stack exchange,提问作者Martin_at_Coventry

火山引擎 最新活动