tf.raw_ops.NcclReduce

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

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

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

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/api_docs/python/tf/raw_ops/NcclReduce