tf.train.ClusterSpec

Представляет кластер как набор «задач», организованных в «работы».

Просмотр псевдонимов

Псевдонимы совместимости для миграции

См. Руководство по миграции для получения дополнительных сведений.

tf.compat.v1.train.ClusterSpec

tf.train.ClusterSpec(
    cluster
)

Объект tf.train.ClusterSpec представляет набор процессов, участвующих в распределённом вычислении TensorFlow. Каждый tf.distribute.Server создаётся в определённом кластере.

Для создания кластера с двумя работами и пятью задачами необходимо указать отображение имён работ на списки сетевых адресов (обычно пары имя_хоста-порт).

cluster = tf.train.ClusterSpec({"worker": ["worker0.example.com:2222",
                                           "worker1.example.com:2222",
                                           "worker2.example.com:2222"],
                                "ps": ["ps0.example.com:2222",
                                       "ps1.example.com:2222"]})

Каждая работа также может быть указана как разреженное отображение от индексов задач к сетевым адресам. Это позволяет настроить сервер, не зная идентификаторов (например) всех других задач работников:

cluster = tf.train.ClusterSpec({"worker": {1: "worker1.example.com:2222"},
                                "ps": ["ps0.example.com:2222",
                                       "ps1.example.com:2222"]})

Аргументы
`cluster`	Словарь, отображающий одно или несколько имён работ на (i) список сетевых адресов или (ii) словарь, отображающий целочисленные индексы задач на сетевые адреса; или протокол `tf.train.ClusterDef` буфера.

Исключения
`TypeError`	Если `cluster` не является словарем, отображающим строки на списки строк, и не является протоколом `tf.train.ClusterDef` буфера.

Атрибуты
`jobs`	Возвращает список имён работ в этом кластере.

Методы

`as_cluster_def`

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

as_cluster_def()

Возвращает tf.train.ClusterDef протокол буфера на основе этого кластера.

`as_dict`

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

as_dict()

Возвращает словарь от имён работ к их задачам.

Для каждой работы, если пространство индексов задач плотное, соответствующее значение будет списком сетевых адресов; в противном случае это будет словарь, отображающий (разреженные) индексы задач на соответствующие адреса.

Возвращает
Словарь, отображающий имена работ на списки или словари, описывающие задачи в этих работах.

`job_tasks`

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

job_tasks(
    job_name
)

Возвращает отображение от идентификатора задачи к адресу в заданной работе.

Примечание: Для обратной совместимости этот метод возвращает список. Если заданная работа была определена с разреженным набором индексов задач, длина этого списка может не отражать количество задач, определённых в этой работе. Используйте метод tf.train.ClusterSpec.num_tasks для поиска количества задач, определённых в конкретной работе.

Аргументы
`job_name`	Имя работы в этом кластере.

Возвращает
Список адресов задач, где индекс в списке соответствует индексу задачи каждой задачи. Список может содержать `None` если работа была определена с разреженным набором индексов задач.

Исключения
`ValueError`	Если `job_name` не является именем работы в этом кластере.

`num_tasks`

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

num_tasks(
    job_name
)

Возвращает количество задач, определённых в заданной работе.

Аргументы
`job_name`	Имя работы в этом кластере.

Возвращает
Количество задач, определённых в заданной работе.

Исключения
`ValueError`	Если `job_name` не является именем работы в этом кластере.

`task_address`

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

task_address(
    job_name, task_index
)

Возвращает адрес заданной задачи в заданной работе.

Аргументы
`job_name`	Имя работы в этом кластере.
`task_index`	Неотрицательное целое число.

Возвращает
Адрес заданной задачи в заданной работе.

Исключения
`ValueError`	Если `job_name` не является именем работы в этом кластере, или задача с индексом `task_index` не определена в этой работе.

`task_indices`

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

task_indices(
    job_name
)

Возвращает список допустимых индексов задач в заданной работе.

Аргументы
`job_name`	Имя работы в этом кластере.

Возвращает
Список допустимых индексов задач в заданной работе.

Исключения
`ValueError`	Если `job_name` не является именем работы в этом кластере, или задача с индексом `task_index` не определена в этой работе.

`bool`

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

__bool__()

`eq`

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

__eq__(
    other
)

Возвращает self==value.

`ne`

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

__ne__(
    other
)

Возвращает self!=value.

`nonzero`

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

__nonzero__()

© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.9/api_docs/python/tf/train/ClusterSpec