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 `nombre_compartido.
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`. Si no lo hace, la ejecución del gráfico no se completará.
input: 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. shared_name: Identificador que se comparte entre operaciones de la misma reducción.
Constantes
Cuerda | OP_NAME | El nombre de esta operación, como lo conoce el motor central de TensorFlow |
Métodos públicos
Métodos heredados
booleano | equals (Object arg0) |
Clase final <?> | getClass () |
En t | hashCode () |
vacío final | notificar () |
vacío final | notifyAll () |
Cuerda | toString () |
vacío final | espera (tiempo arg0, arg1 int) |
vacío final | espera (arg0 largo) |
vacío final | wait () |
abstracta ExecutionEnvironment | env () Devuelve el entorno de ejecución en el que se creó esta operación. |
abstracta Operación |
Constantes
OP_NAME pública final static String
El nombre de esta operación, como lo conoce el motor central de TensorFlow
Métodos públicos
pública de salida <T> asOutput ()
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.
public static NcclAllReduce <T> crear ( Alcance alcance, operando <T> de entrada, la reducción de cuerdas, numDevices largas, Cadena sharedName)
Método de fábrica para crear una clase que envuelva una nueva operación NcclAllReduce.
Parámetros
alcance | alcance actual |
---|
Devoluciones
- una nueva instancia de NcclAllReduce