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