tf.distribute.InputContext

Класс, оборачивающий информацию, необходимую для функции ввода.

Псевдонимы

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

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

tf.compat.v1.distribute.InputContext

tf.distribute.InputContext(
    num_input_pipelines=1, input_pipeline_id=0, num_replicas_in_sync=1
)

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

Аргументы
`num_input_pipelines`	количество каналов ввода в кластере.
`input_pipeline_id`	идентификатор текущего канала ввода, должен быть целым числом в диапазоне [0,`num_input_pipelines`).
`num_replicas_in_sync`	количество реплик, которые синхронизированы.

Атрибуты
`input_pipeline_id`	Возвращает идентификатор канала ввода.
`num_input_pipelines`	Возвращает количество каналов ввода.
`num_replicas_in_sync`	Возвращает количество вычислительных реплик, которые синхронизированы.

Методы

`get_per_replica_batch_size`

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

get_per_replica_batch_size(
    global_batch_size
)

Возвращает размер пакета на реплику.

Аргументы
`global_batch_size`	глобальный размер пакета, который должен быть кратен `num_replicas_in_sync`.

Возвращает
размер пакета на реплику.

Исключения
`ValueError`	если `global_batch_size` не кратен `num_replicas_in_sync`.

© 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/api_docs/python/tf/distribute/InputContext