tf.distribute.InputContext

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

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

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

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

tf.compat.v1.distribute.InputContext, `tf.compat.v2.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`.

© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r1.15/api_docs/python/tf/distribute/InputContext