find
和TensorFlow
似乎都不支持循环计算图,循环元素实现为具有缓冲和展开的递归细胞(RNN / LSTM细胞),但这种限制主要与反向传播的计算。我不需要计算反向传播,只需要前向传播。
有没有办法忽略这个限制,或者只是为了分解非循环组件中的任意计算图?
答案 0 :(得分:3)
TensorFlow 支持支持循环计算图。 tf.while_loop()
函数允许您为循环的条件和主体指定带有任意子图的while循环,运行时将并行执行循环。 tf.scan()
函数是一个更高级别的API,类似于Theano的theano.scan()
函数。两者都允许你循环动态大小的张量。