AllToAll

classe finale pubblica AllToAll

Un'operazione per scambiare dati tra repliche TPU.

Su ciascuna replica, l'input viene suddiviso in blocchi "split_count" lungo "split_dimension" e inviato alle altre repliche in base a group_assignment. Dopo aver ricevuto i blocchi "split_count" - 1 da altre repliche, concateniamo i blocchi lungo "concat_dimension" come output.

Ad esempio, supponiamo che siano presenti 2 repliche TPU: la replica 0 riceve input: `[[A, B]]` la replica 1 riceve input: `[[C, D]]`

group_assignment=`[[0, 1]]` concat_dimension=0 split_dimension=1 split_count=2

output della replica 0: `[[A], [C]]` output della replica 1: `[[B], [D]]`

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 TType > AllToAll <T>	create ( ambito ambito , operando <T> input, operando < TInt32 > groupAssignment, concatDimension lungo, splitDimension lungo, splitCount lungo) Metodo factory per creare una classe che racchiude una nuova operazione AllToAll.
Uscita <T>	produzione () Il risultato scambiato.

Metodi ereditati

Dalla classe org.tensorflow.op.RawOp

booleano finale	è uguale a (Oggetto oggetto)
finale int	codice hash ()
Operazione	operazione () Restituisce questa unità di calcolo come una singola `Operation` .
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.op.Op

ambiente di esecuzione astratto	ambiente () Restituisce l'ambiente di esecuzione in cui è stata creata questa operazione.
Operazione astratta	operazione () Restituisce questa unità di calcolo come una singola `Operation` .

Dall'interfaccia org.tensorflow.Operand

risultato astratto <T>	comeuscita () Restituisce l'handle simbolico del tensore.
astratto T	come tensore () Restituisce il tensore in questo operando.
forma astratta	forma () Restituisce la forma (possibilmente parzialmente conosciuta) del tensore a cui fa riferimento l' `Output` di questo operando.
Classe astratta<T>	tipo () Restituisce il tipo tensore di questo operando

Dall'interfaccia org.tensorflow.ndarray.Shaped

astratto int	rango ()
forma astratta	forma ()
astratto lungo	misurare () Calcola e restituisce la dimensione totale di questo contenitore, in numero di valori.

Costanti

Stringa finale statica pubblica OP_NAME

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

Valore costante: "AllToAll"

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 AllToAll <T> create ( ambito ambito , operando <T> input, operando < TInt32 > groupAssignment, Long concatDimension, Long splitDimension, Long splitCount)

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

Parametri

scopo	ambito attuale
ingresso	L'input locale alla somma.
groupAssignment	Un tensore int32 con forma [num_groups, num_replicas_per_group]. "group_assignment[i]" rappresenta gli ID di replica nell'i-esimo sottogruppo.
concatDimension	Il numero della dimensione da concatenare.
dimensione divisa	Il numero della dimensione da dividere.
splitCount	Il numero di suddivisioni, questo numero deve essere uguale alla dimensione del sottogruppo (group_assignment.get_shape()[1])

ritorna

una nuova istanza di AllToAll

Uscita pubblica <T> uscita ()

Il risultato scambiato.