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