NcclReduce

Clase public final NcclReduce

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>
crear ( Alcance alcance, Iterable < Operando <T >> de entrada, la reducción String)
Método de fábrica para crear una clase que envuelva una nueva operación NcclReduce.
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

Valor constante: "NcclReduce"

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

pública de salida <T> de datos ()