使用Tensorflow或Theano的循环计算图

时间:2016-05-31 16:46:22

标签: tensorflow theano lstm recurrent-neural-network

findTensorFlow似乎都不支持循环计算图,循环元素实现为具有缓冲和展开的递归细胞(RNN / LSTM细胞),但这种限制主要与反向传播的计算。我不需要计算反向传播,只需要前向传播。

有没有办法忽略这个限制,或者只是为了分解非循环组件中的任意计算图?

1 个答案:

答案 0 :(得分:3)

TensorFlow 支持支持循环计算图。 tf.while_loop()函数允许您为循环的条件和主体指定带有任意子图的while循环,运行时将并行执行循环。 tf.scan()函数是一个更高级别的API,类似于Theano的theano.scan()函数。两者都允许你循环动态大小的张量。