tf.raw_ops.NcclAllReduce

Возвращает тензор, содержащий результат редукции всех входных тензоров.

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

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

Подробнее см. в руководстве по миграции.

tf.compat.v1.raw_ops.NcclAllReduce

tf.raw_ops.NcclAllReduce(
    input, reduction, num_devices, shared_name, name=None
)

Возвращает тензор, содержащий результат редукции всех входных тензоров, переданных операциям с одинаковым значением `shared_name.

Граф следует построить так, чтобы, если одна операция выполняется со значением shared_name c, то num_devices операций выполнялись со значением shared_name c. Если этого не сделать, выполнение графа не завершится.

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

Аргументы
`input`	Объект типа `Tensor`. Должен иметь один из следующих типов: `half`, `float32`, `float64`, `int32`, `int64`.
`reduction`	Объект типа `string` из: `"min", "max", "prod", "sum"`.
`num_devices`	Объект типа `int`.
`shared_name`	Объект типа `string`.
`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/NcclAllReduce