tf.raw_ops.NcclReduce

Сводит input из num_devices с помощью reduction к одному устройству.

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

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

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

tf.compat.v1.raw_ops.NcclReduce

tf.raw_ops.NcclReduce(
    input, reduction, name=None
)

Сводит input из num_devices с помощью reduction к одному устройству.

Граф должен быть построен таким образом, чтобы все входные данные имели действительное назначение устройства, и операция сама назначена одному из этих устройств.

input: Вход для сведения. data: значение сведения по всем num_devices устройствам. reduction: операция сведения.

Аргументы
`input`	Список из как минимум 1 объекта `Tensor` с одинаковым типом в: `half`, `float32`, `float64`, `int32`, `int64`.
`reduction`	Объект `string` из: `"min", "max", "prod", "sum"`.
`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/versions/r2.9/api_docs/python/tf/raw_ops/NcclReduce