通过属性将设备分配给Tensorflow中的操作

时间:2018-04-02 23:47:32

标签: python tensorflow

有没有办法在创建操作后分配设备?所以不要使用 tf.device(u'/ gpu:1'),例如我们可以使用 op.device = u'/ gpu:1'(显然,这不起作用,但想知道是否有类似的东西)

注意:操作是张量流操作, type(op)返回 tensorflow.python.framework.ops.Operation

谢谢

0 个答案:

没有答案