tf.tpu.cross_replica_sum

Суммирование входного тензора по репликам в соответствии с group_assignment.

Основные псевдонимы

`tf.contrib.tpu.cross_replica_sum`

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

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

tf.tpu.cross_replica_sum(
    x, group_assignment=None, name=None
)

Аргументы
`x`	Местный тензор для суммирования.
`group_assignment`	Необязательные 2d списки int32 с формой [num_groups, num_replicas_per_group]. `group_assignment[i]` представляет идентификаторы реплик в i-й подгруппе.
`name`	Необязательное имя операции.

Возвращаемое значение
Тензор `Tensor`, который суммируется по репликам.

© 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/tpu/cross_replica_sum