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`.

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