CollectiveAllToAllV2

classe finale pubblica CollectiveAllToAllV2

Scambia reciprocamente più tensori di identico tipo e forma.

"is_stateless" significa che ogni operazione non necessita di dipendenze di controllo da altre operazioni collettive. In questo caso, per distinguere i gruppi collettivi dovrebbero essere utilizzate chiavi univoche in fase di esecuzione (ad esempio `instance_key`).

Classi nidificate

classe CollectiveAllToAllV2.Options Attributi facoltativi per CollectiveAllToAllV2

Metodi pubblici

Uscita <T>
comeuscita ()
Restituisce la maniglia simbolica di un tensore.
CollectiveAllToAllV2.Options statico
comunicazioneHint (Stringa comunicazioneHint)
statico <T estende il numero> CollectiveAllToAllV2 <T>
create ( ambito ambito , Operando <T> input, Operando <Integer> groupSize, Operando <Integer> groupKey, Operando <Integer> exampleKey, Iterable< Operando <?>> orderingToken, Opzioni... opzioni)
Metodo factory per creare una classe che racchiude una nuova operazione CollectiveAllToAllV2.
Uscita <T>
dati ()
CollectiveAllToAllV2.Options statico
isStateless (Booleano isStateless)
CollectiveAllToAllV2.Options statico
timeoutSecondi (TimeoutSecondi mobile)

Metodi ereditati

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 CollectiveAllToAllV2.Options communicationsHint (Stringa communicationsHint)

public static CollectiveAllToAllV2 <T> create ( Scope scope, Operando <T> input, Operando <Integer> groupSize, Operando <Integer> groupKey, Operando <Integer> exampleKey, Iterable< Operando <?>> orderingToken, Opzioni... opzioni)

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

Parametri
scopo ambito attuale
opzioni trasporta valori di attributi opzionali
ritorna
  • una nuova istanza di CollectiveAllToAllV2

output pubblico <T> dati ()

public static CollectiveAllToAllV2.Options isStateless (Boolean isStateless)

public static CollectiveAllToAllV2.Options timeoutSeconds (Float timeoutSeconds)