Указывает устройство для операций, созданных/выполненных в данном контексте.
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
Имя устройства, которое следует использовать в контексте.
Возвращаемое значение
Менеджер контекста, который определяет устройство по умолчанию для вновь созданных операций.
