NcclAllReduce

classe finale pubblica NcclAllReduce

Restituisce un tensore contenente la riduzione su tutti i tensori di input.

Restituisce un tensore contenente la riduzione tra tutti i tensori di input passati a ops all'interno dello stesso `shared_name.

Il grafico dovrebbe essere costruito in modo che se un'operazione viene eseguita con il valore shared_name `c`, le operazioni `num_devices` verranno eseguite con il valore shared_name `c`. In caso contrario, l'esecuzione del grafico non verrà completata.

input: l'input per la riduzione dati: il valore della riduzione su tutti i dispositivi "num_devices". riduzione: l'operazione di riduzione da eseguire. num_devices: il numero di dispositivi che partecipano a questa riduzione. shared_name: identificatore condiviso tra le operazioni della stessa riduzione.

Metodi pubblici

Uscita <T>	comeuscita () Restituisce la maniglia simbolica di un tensore.
statico <T estende Numero> NcclAllReduce <T>	create (ambito ambito , input operando <T>, riduzione stringa, numDevices lunghi, stringa sharedName) Metodo factory per creare una classe che racchiude una nuova operazione NcclAllReduce.
Uscita <T>	dati ()

Metodi ereditati

Dalla classe org.tensorflow.op.PrimitiveOp

booleano finale	è uguale a (Oggetto oggetto)
finale int	codice hash ()
Operazione	operazione () Restituisce l' Operation sottostante
stringa finale	accordare ()

Dalla classe java.lang.Object

booleano	è uguale a (Oggetto arg0)
Classe finale<?>	getClass ()
int	codice hash ()
vuoto finale	notificare ()
vuoto finale	notificaTutti ()
Corda	accordare ()
vuoto finale	attendere (lungo arg0, int arg1)
vuoto finale	aspetta (lungo arg0)
vuoto finale	Aspettare ()

Dall'interfaccia org.tensorflow.Operand

Output astratto <T>

comeuscita () Restituisce la maniglia simbolica di un tensore.

Metodi pubblici

Uscita pubblica <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 NcclAllReduce <T> create ( ambito ambito , operando <T> input, riduzione stringa, numDevices lunghi, stringa sharedName)

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

Parametri

scopo	ambito attuale

ritorna

• una nuova istanza di NcclAllReduce

output pubblico <T> dati ()