Reduce la `entrada` de` num_devices` usando `reducción` a un solo dispositivo.
Reduce la `entrada` de` num_devices` usando `reducción` a un solo dispositivo.
El gráfico debe construirse de modo que todas las entradas tengan una asignación de dispositivo válida, y al operador en sí se le asigne uno de estos dispositivos.
entrada: La entrada a la reducción. datos: el valor de la reducción en todos los dispositivos `num_devices`. reducción: la operación de reducción a realizar.
Constantes
Cuerda | OP_NAME | El nombre de esta operación, como lo conoce el motor central de TensorFlow |
Métodos públicos
Salida <T> | asOutput () Devuelve el identificador simbólico del tensor. |
estática <T se extiende TNumber > NcclReduce <T> | |
Salida <T> | datos () |
Métodos heredados
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 NcclReduce <T> crear ( Alcance alcance, Iterable < operando <T >> de entrada, la reducción de cuerdas)
Método de fábrica para crear una clase que envuelva una nueva operación NcclReduce.
Parámetros
alcance | alcance actual |
---|
Devoluciones
- una nueva instancia de NcclReduce