tf.device

Задаёт устройство для операций, создаваемых или выполняемых в этом контексте.

tf.device(
    device_name
) -> ContextManager[None]

Используется в блокнотах

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