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.I использовать with tf.device
здесь, чтобы проиллюстрировать, что ": 0" не означает, что переменная лежит на конкретном устройстве. Так что значение ": 0" в имени переменной (foo и бар в этом примере)?