NcclReduce

classe finale pubblica NcclReduce

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

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

Il grafico dovrebbe essere costruito in modo che tutti gli input 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.

Costanti

Corda OP_NAME Il nome di questa operazione, come noto al motore principale di TensorFlow

Metodi pubblici

Uscita <T>
comeuscita ()
Restituisce l'handle simbolico del tensore.
statico <T estende TNumber > NcclReduce <T>
create (ambito dell'ambito , input Iterable< operando <T>>, riduzione della stringa)
Metodo factory per creare una classe che racchiude una nuova operazione NcclReduce.
Uscita <T>
dati ()

Metodi ereditati

Costanti

Stringa finale statica pubblica OP_NAME

Il nome di questa operazione, come noto al motore principale di TensorFlow

Valore costante: "NcclReduce"

Metodi pubblici

Uscita pubblica <T> asOutput ()

Restituisce l'handle simbolico del 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 (ambito ambito , Iterable < operando <T>> input, riduzione stringa)

Metodo factory per creare una classe che racchiude una nuova operazione NcclReduce.

Parametri
scopo ambito attuale
ritorna
  • una nuova istanza di NcclReduce

output pubblico <T> dati ()