【发布时间】:2016-12-02 05:51:11
【问题描述】:
import tensorflow as tf
with tf.device('/gpu:0'):
foo = tf.Variable(1, name='foo')
assert foo.name == "foo:0"
with tf.device('/gpu:1'):
bar = tf.Variable(1, name='bar')
assert bar.name == "bar:0"
上面的代码返回true。我这里用with tf.device来说明“:0”并不表示变量位于特定的设备上。那么变量名中的“:0”是什么意思(本例中为 foo 和 bar)?
【问题讨论】:
-
断言返回真吗?你读过
tensorflow.Variable类的文档吗? -
@cricket_007 是的,它返回true,并且在tensorflow doc,你可以找到一些类似的代码,但是文档从来没有解释“:0”的含义。
-
明白了。我只是在阅读源代码。不过不容易发现。
-
据我所知,它的意思是“第 0 次迭代后的变量栏”,但我不是经验丰富的 TF 用户
标签: python tensorflow