tf.compat.v2.DeviceSpec

Представляет (возможно, частичное) описание устройства TensorFlow.

tf.compat.v2.DeviceSpec(
    job=None, replica=None, task=None, device_type=None, device_index=None
)

DeviceSpec используются в TensorFlow для описания места хранения состояния и выполнения вычислений. Использование DeviceSpec позволяет анализировать строки спецификаций устройств для проверки их корректности, объединения или программирования их композиции.

Пример:

# Place the operations on device "GPU:0" in the "ps" job.
device_spec = DeviceSpec(job="ps", device_type="GPU", device_index=0)
with tf.device(device_spec):
  # Both my_var and squared_var will be placed on /job:ps/device:GPU:0.
  my_var = tf.Variable(..., name="my_variable")
  squared_var = tf.square(my_var)

Если DeviceSpec частично указано, оно будет объединено с другими DeviceSpec в соответствии со областью, в которой оно определено. DeviceSpec компоненты, определенные во внутренних областях, имеют приоритет над компонентами, определенными во внешних областях.

with tf.device(DeviceSpec(job="train", )):
  with tf.device(DeviceSpec(job="ps", device_type="GPU", device_index=0):
    # Nodes created here will be assigned to /job:ps/device:GPU:0.
  with tf.device(DeviceSpec(device_type="GPU", device_index=1):
    # Nodes created here will be assigned to /job:train/device:GPU:1.

DeviceSpec состоит из 5 компонентов — каждый из которых необязательно указан:

• Job: Имя задания.
• Replica: Индекс реплики.
• Task: Индекс задачи.
• Тип устройства: Строка типа устройства (например, "CPU" или "GPU").
• Индекс устройства: Индекс устройства.

Методы

from_string

Просмотреть исходный код

@classmethod
from_string(
    spec
)

Создать DeviceSpec из строки.

make_merged_spec

Просмотреть исходный код

make_merged_spec(
    dev
)

Возвращает новый DeviceSpec, который включает dev.

При объединении спецификаций dev будет иметь приоритет над текущей спецификацией. Например:

first_spec = tf.DeviceSpec(job=0, device_type="CPU")
second_spec = tf.DeviceSpec(device_type="GPU")
combined_spec = first_spec.make_merged_spec(second_spec)

эквивалентно:

combined_spec = tf.DeviceSpec(job=0, device_type="GPU")

parse_from_string

Просмотреть исходный код

parse_from_string(
    spec
)

Разбор имени DeviceSpec в его компоненты.

Изменение поведения 2.x: В TensorFlow 1.x эта функция изменяет собственное состояние и возвращает себя. В 2.x DeviceSpec неизменяемы, и эта функция вернет DeviceSpec, содержащий спецификацию.

Рекомендуется:

# my_spec and my_updated_spec are unrelated.
my_spec = tf.DeviceSpec.from_string("/CPU:0")
my_updated_spec = tf.DeviceSpec.from_string("/GPU:0")
with tf.device(my_updated_spec):
  ...

Будет работать в 1.x и 2.x (хотя устарело в 2.x):

my_spec = tf.DeviceSpec.from_string("/CPU:0")
my_updated_spec = my_spec.parse_from_string("/GPU:0")
with tf.device(my_updated_spec):
  ...

Не будет работать в 2.x:

my_spec = tf.DeviceSpec.from_string("/CPU:0")
my_spec.parse_from_string("/GPU:0")  # <== Will not update my_spec
with tf.device(my_spec):
  ...

В общем случае, DeviceSpec.from_string должен полностью заменить DeviceSpec.parse_from_string, и DeviceSpec.replace должен полностью заменить непосредственное изменение атрибутов.

replace

Просмотреть исходный код

replace(
    **kwargs
)

Удобный метод для создания нового DeviceSpec путем переопределения полей.

Например:

my_spec = DeviceSpec=(job="my_job", device="CPU")
my_updated_spec = my_spec.replace(device="GPU")
my_other_spec = my_spec.replace(device=None)

to_string

Просмотреть исходный код

to_string()

Возвращает строковое представление этого DeviceSpec.

__eq__

Просмотреть исходный код

__eq__(
    other
)

Проверяет, является ли DeviceSpec other таким же, как текущий экземпляр, т.е. имеют

одинаковые значения для всех внутренних полей.