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