Указывает устройство для операций, созданных/выполненных в данном контексте.
tf.device(
device_name
)
Эта функция указывает устройство, которое будет использоваться для операций, созданных/выполненных в конкретном контексте. Вложенные контексты будут наследовать и также создавать/выполнять свои операции на указанном устройстве. Если конкретное устройство не требуется, следует воздержаться от использования этой функции, чтобы устройство можно было автоматически назначить. В общем случае использование этой функции необязательно. device_name может быть полностью указано, как в "/job:worker/task:1/device:cpu:0", или частично указано, содержа в себе только подмножество полей, разделенных "/". Любые указанные поля будут переопределять аннотации устройств из внешних областей.
Например:
with tf.device('/job:foo'):
# ops created here have devices with /job:foo
with tf.device('/job:bar/task:0/device:gpu:2'):
# ops created here have the fully specified device above
with tf.device('/device:gpu:1'):
# ops created here have the device '/job:foo/device:gpu:1'
Аргументы
device_name
Имя устройства, которое нужно использовать в контексте.
Возвращаемое значение
Менеджер контекста, который указывает на устройство по умолчанию для новых создаваемых операций.
