标签: tensorflow
假设我有两个TF图G_1,G_2,我想取G_1的输出并将其输入G_2。我的假设是,为此,我们在G_1的Feed上输入GPU,然后取G_1的输出并将其输入G_2,然后输出到CPU。
G_1
G_2
有没有办法在GPU上提供两个图形?
答案 0 :(得分:2)
通常,您不必明确指定CPU或GPU。 TensorFlow使用您的第一个GPU(如果有的话)进行尽可能多的操作。
G_1的输出将进入G-2,这也将在GPU上执行。
G-2
资源:Tensorflow Basic Usage