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`.

© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.4/api_docs/python/tf/raw_ops/NcclReduce