tf.compat.v1.tpu.cross_replica_sum
Суммирует входной тензор по репликам в соответствии с group_assignment.
tf.compat.v1.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, который суммируется по репликам. |
© 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/versions/r2.9/api_docs/python/tf/compat/v1/tpu/cross_replica_sum
