tensorflow - 使用 Tensorflow 或 Theano 的循环计算图

标签 tensorflow theano lstm recurrent-neural-network

TensorFlowTheano 似乎都不支持循环计算图,循环元素被实现为具有缓冲和展开的循环单元(RNN/LSTM 单元),但是这个限制主要与反向传播的计算有关。我没有特别需要计算反向传播,只是需要计算前向传播。

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

最佳答案

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

关于tensorflow - 使用 Tensorflow 或 Theano 的循环计算图,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/37551389/

相关文章:

python - 如何计算 Tensorflow 中矩阵对角线的总和?

python - 如何避免 Theano 计算梯度趋向于 NaN

python - Python Theano 中的二进制输出神经网络

python - 如何关闭/打开 LSTM 层?

tensorflow - 如何在同一张图片中定位多个物体?

tensorflow - tensorflow 上的线性回归模型无法学习偏差

python - 使用 Tensorflow Keras 将 CNN 与 LSTM 相结合

python-3.x - `return_sequences = False` 在 pytorch LSTM 中等效

python - Tensorflow 可以用于多元函数的全局最小化吗?

python - 如何让keras+theano使用>1个核心