NcclReduce

public final class NcclReduce

Riduce "input" da "num_devices" usando "reduction" a un singolo dispositivo.

Riduce "input" da "num_devices" usando "reduction" a un singolo dispositivo.

Il grafico dovrebbe essere costruito in modo che tutti gli ingressi abbiano un'assegnazione di dispositivo valida e all'operazione stessa venga assegnato uno di questi dispositivi.

input: l'input per la riduzione. data: il valore della riduzione su tutti i dispositivi `num_devices`. riduzione: l'operazione di riduzione da eseguire.

Metodi pubblici

Uscita <T>
asOutput ()
Restituisce la maniglia simbolica di un tensore.
static <T extends Number> NcclReduce <T>
create ( ambito ambito, input < Operando <T>> Iterabile, riduzione delle stringhe)
Metodo Factory per creare una classe che avvolge una nuova operazione NcclReduce.
Uscita <T>
dati ()

Metodi ereditati

Metodi pubblici

output pubblico <T> asOutput ()

Restituisce la maniglia simbolica di un tensore.

Gli input per le operazioni TensorFlow sono output di un'altra operazione TensorFlow. Questo metodo viene utilizzato per ottenere un handle simbolico che rappresenta il calcolo dell'input.

public static NcclReduce <T> create ( Scope scope, Iterable < Operand <T>> input, String reduction)

Metodo Factory per creare una classe che avvolge una nuova operazione NcclReduce.

Parametri
scopo ambito attuale
ritorna
  • una nuova istanza di NcclReduce

output pubblico <T> dati ()