NcclReduce

clase final pública 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 que a la operación 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

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 > NcclReduce <T>
crear (alcance alcance , entrada Iterable< operando <T>>, reducción de cadena)
Método de fábrica para crear una clase que envuelve una nueva operación NcclReduce.
Salida <T>
datos ()

Métodos heredados

Constantes

Cadena final estática pública OP_NAME

El nombre de esta operación, como lo conoce el motor central de TensorFlow.

Valor constante: "NcclReduce"

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 NcclReduce <T> (alcance del alcance , entrada Iterable < operando <T>>, reducción de cadena)

Método de fábrica para crear una clase que envuelve una nueva operación NcclReduce.

Parámetros
alcance alcance actual
Devoluciones
  • una nueva instancia de NcclReduce

datos públicos de salida <T> ()