我想生成一批具有固定形状 [ batch_size
, seq_lenth
] 的序列。因此我的代码:
output, _ = tf.nn.ctc_beam_search_decoder(
logits,
tf.ones([batch_size])*seq_length,
merge_repeated=False
)
output = tf.sparse_to_dense(output[0], default_value=end_id)
其中logits
是dynamic_rnn
形状为 [ max_time
, batch_size
, cell_output_size
] 的输出。我设置seq_length
为40
. 但结果output
不是形状 [ batch_size
, seq_length
]。鉴于它的形状是动态的,我无法将它填充到固定长度。那么我应该怎么做才能得到固定大小的输出呢?提前致谢!