NcclReduce

clase final pública NcclReduce

Reduce `input` de `num_devices` usando `reduction` a un solo dispositivo.

Reduce `input` de `num_devices` usando `reduction` 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 propio operador 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.

Métodos públicos

Salida <T>
como salida ()
Devuelve el identificador simbólico de un tensor.
estático <T extiende Número> NcclReduce <T>
crear ( ámbito de alcance, iterable< operando <T>> entrada, 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

Métodos públicos

salida pública <T> como salida ()

Devuelve el identificador simbólico de un tensor.

Las entradas de 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> create ( Scope scope, Iterable< Operand <T>> entrada, Cadena de reducción)

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

salida pública <T> datos ()