张量流中可变范围的循环

时间:2016-08-01 10:14:11

标签: tensorflow placeholder lstm

我在tensorflow中实现了LSTM的代码,每个批处理都有不同的长度。为了将数据提供给图表,我需要有可变长度的输入列表。

我希望下面的循环在tensorflow中有一个变量范围。我可以在创建图形时将其初始化为特定值,但是无法进行任何修改。我试图使用全局列表的长度作为外部参数,但它不起作用。

self._train_data = []
for _ in range( variable_length ):
    placeholder_input = tf.placeholder(tf.float32, shape=[self._batch_size, self._input_size])    
    self._train_data.append(placeholder_input)

1 个答案:

答案 0 :(得分:1)

TensorFlow seq2seq tutorial讨论了可变长度的问题,并包含处理它们的代码。它是在tf.while_loop之前编写的,现在它允许您更好地处理多个长度。但一般规则仍然适用,所以看看那里!