TensorFlow
和 Theano
似乎都不支持循环计算图,循环元素被实现为具有缓冲和展开的循环单元(RNN/LSTM 单元),但是这个限制主要与反向传播的计算有关。我没有特别需要计算反向传播,只是需要计算前向传播。
有没有办法忽略这个限制,或者只是分解非循环组件中的任意计算图?
最佳答案
TensorFlow 确实支持循环计算图。 tf.while_loop()
函数允许您为条件和循环体指定带有任意子图的 while 循环,运行时将并行执行该循环。 tf.scan()
function 是一个更高级别的 API,类似于 Theano 的 theano.scan()
功能。两者都允许您循环动态大小的张量。
关于tensorflow - 使用 Tensorflow 或 Theano 的循环计算图,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/37551389/