tf.raw_ops.CrossReplicaSum

Операция суммирования входных данных по реплицированным экземплярам TPU.

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

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

Дополнительные сведения см. в руководстве по миграции https://www.tensorflow.org/guide/migrate.

tf.compat.v1.raw_ops.CrossReplicaSum

tf.raw_ops.CrossReplicaSum(
    input, group_assignment, name=None
)

Каждый экземпляр предоставляет свой собственный вход.

Например, предположим, что есть 8 экземпляров TPU: [A, B, C, D, E, F, G, H]. Передача group_assignment=[[0,2,4,6],[1,3,5,7]] устанавливает A, C, E, G в группу 0 и B, D, F, H в группу 1. Таким образом, мы получаем следующие выходные данные: [A+C+E+G, B+D+F+H, A+C+E+G, B+D+F+H, A+C+E+G, B+D+F+H, A+C+E+G, B+D+F+H].

Аргументы
`input`	`Tensor`. Должен быть одним из следующих типов: `half`, `bfloat16`, `float32`, `float64`, `int32`, `uint32`. Местный вход для суммирования.
`group_assignment`	`Tensor` типа `int32`. Массив int32 с формой [num_groups, num_replicas_per_group]. `group_assignment[i]` представляет идентификаторы реплик в i-й подгруппе.
`name`	Имя операции (необязательно).

Возвращаемые значения
`Tensor`. Имеет тот же тип, что и `input`.

© 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/raw_ops/CrossReplicaSum