Genera un tensor que contiene la reducción en todos los tensores de entrada.
Genera un tensor que contiene la reducción en todos los tensores de entrada pasados a operaciones dentro del mismo `shared_name.
El gráfico debe construirse de modo que si una operación se ejecuta con el valor de nombre compartido `c`, entonces las operaciones de `num_devices` se ejecutarán con el valor de nombre compartido `c`. De lo contrario, la ejecución del gráfico no se completará.
entrada: la entrada a los datos de reducción: el valor de la reducción en todos los dispositivos `num_devices`. reducción: la operación de reducción a realizar. num_devices: la cantidad de dispositivos que participan en esta reducción. nombre_compartido: Identificador compartido entre operaciones de la misma reducción.
Constantes
| Cadena | OP_NOMBRE | El nombre de esta operación, como lo conoce el motor central de TensorFlow. |
Métodos públicos
| Salida <T> | como salida () Devuelve el identificador simbólico del tensor. |
| estático <T extiende TNumber > NcclAllReduce <T> | crear (ámbito de alcance , entrada de operando <T>, reducción de cadena, número largo de dispositivos, nombre compartido de cadena) Método de fábrica para crear una clase que envuelve una nueva operación NcclAllReduce. |
| Salida <T> | datos () |
Métodos heredados
| booleano | es igual (Objeto arg0) |
| Clase final<?> | obtenerclase () |
| En t | código hash () |
| vacío final | notificar () |
| vacío final | notificar a todos () |
| Cadena | Encadenar () |
| vacío final | esperar (arg0 largo, int arg1) |
| vacío final | espera (arg0 largo) |
| vacío final | esperar () |
| entorno de ejecución abstracto | entorno () Devuelve el entorno de ejecución en el que se creó esta operación. |
| operación abstracta |
| Salida abstracta <T> | como salida () Devuelve el identificador simbólico del tensor. |
| resumen T | como Tensor () Devuelve el tensor en este operando. |
| forma abstracta | |
| clase abstracta<T> | tipo () Devuelve el tipo de tensor de este operando. |
Constantes
Cadena final estática pública OP_NAME
El nombre de esta operación, como lo conoce el motor central de TensorFlow.
Métodos públicos
Salida pública <T> como Salida ()
Devuelve el identificador simbólico del tensor.
Las entradas a las operaciones de TensorFlow son salidas de otra operación de TensorFlow. Este método se utiliza para obtener un identificador simbólico que representa el cálculo de la entrada.
creación estática pública NcclAllReduce <T> (alcance del alcance , entrada del operando <T>, reducción de cadena, número largo de dispositivos, nombre compartido de cadena)
Método de fábrica para crear una clase que envuelve una nueva operación NcclAllReduce.
Parámetros
| alcance | alcance actual |
|---|
Devoluciones
- una nueva instancia de NcclAllReduce