似乎 tensorflow 不支持双向 RNN 的可变批量大小。在此示例中,sequence_length
与 绑定batch_size
,这是一个 Python 整数:
_seq_len = tf.fill([batch_size], tf.constant(n_steps, dtype=tf.int64))
outputs, state1,state2 = rnn.bidirectional_rnn(rnn_fw_cell, rnn_bw_cell, input,
dtype="float",
sequence_length=_seq_len)
如何使用不同的批量大小进行训练和测试?