Tensorflow RNN如何创建具有各种批处理大小的零状态？

Question

在此问题How do I set TensorFlow RNN state when state_is_tuple=True?中：接受的答案将初始化初始状态，如下所示：

state_placeholder = tf.placeholder(tf.float32, [num_layers, 2, batch_size, state_size])

我认为这需要特定的批次大小，而我现在拥有的是：

inputSeq = tf.placeholder(tf.float32, [None, seqLength, observationDim], name='input_seq')
outputs, final_state = tf.nn.dynamic_rnn(cell, inputSeq, initial_state=initialState)

我希望此initialState处于零状态，并且可以配置，因为inputSeq的批处理大小可能会有所不同。但是，cell.zero_state不接受None作为批处理大小。有什么解决方法吗？

Answer 1

cell.zero_state接受标量张量。

通过tf.shape获取占位符的批量大小，然后完成： B = tf.shape(state_placeholder)[0] # the batch size scalar tensor initial_state = cell.zero_state(B)