Tensorflow 是否可以预编译图形,然后并行运行?

Tensorflow 是否可以预编译图形,然后并行运行?,tensorflow,Tensorflow,我遵循递归神经网络的CS224D代码(参见:Pset 3解决方案) 代码的工作原理是为每个训练示例编译一个新的图,因为每个树都有不同的形状,但这对于运行小批量等来说是次优的。我有一个类似的递归图,但输入量较大,因此我需要一个更可并行化的解决方案 我的想法是预编译图形,并将它们与数据一起异步加载到GPU,类似于队列的工作方式。然后卸载小批量结果,计算组合梯度,然后反向传播。我的想法是,我可以考虑变量输入,也可以利用GPU(吃我的蛋糕,也吃它) 这样做可行吗

我遵循递归神经网络的CS224D代码(参见:Pset 3解决方案)

代码的工作原理是为每个训练示例编译一个新的图,因为每个树都有不同的形状,但这对于运行小批量等来说是次优的。我有一个类似的递归图,但输入量较大,因此我需要一个更可并行化的解决方案

我的想法是预编译图形,并将它们与数据一起异步加载到GPU,类似于队列的工作方式。然后卸载小批量结果,计算组合梯度,然后反向传播。我的想法是,我可以考虑变量输入,也可以利用GPU(吃我的蛋糕,也吃它)

这样做可行吗