CollectivePermute

classe finale pubblica CollectivePermute

Un'operazione per permutare i tensori tra le istanze TPU replicate.

Ogni istanza fornisce il proprio input.

Ad esempio, supponiamo che siano presenti 4 istanze TPU: "[A, B, C, D]". Passando source_target_pairs=`[[0,1],[1,2],[2,3],[3,0]]` si ottengono gli output: "[D, A, B, C]".

Metodi pubblici

Uscita <T>	comeuscita () Restituisce la maniglia simbolica di un tensore.
statico <T> CollectivePermute <T>	create ( ambito ambito , operando <T> input, operando <Integer> sourceTargetPairs) Metodo factory per creare una classe che racchiude una nuova operazione CollectivePermute.
Uscita <T>	produzione () L'input permutato.

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 CollectivePermute <T> create ( ambito ambito , operando <T> input, operando <Integer> sourceTargetPairs)

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

Parametri

scopo	ambito attuale
ingresso	L'ingresso locale da permutare. Attualmente supporta solo float e bfloat16.
sourceTargetPairs	Un tensore con forma [num_pairs, 2].

ritorna

• una nuova istanza di CollectivePermute

Uscita pubblica <T> uscita ()

L'input permutato.