CollectiveGather

public final class CollectiveGather

Accumula reciprocamente più tensori di tipo e forma identici.

Classi annidate

classe CollectiveGather.Options Attributi opzionali per CollectiveGather

Metodi pubblici

Uscita <T>
asOutput ()
Restituisce la maniglia simbolica di un tensore.
static CollectiveGather.Options
communicationHint (String communicationHint)
static <T extends Number> CollectiveGather <T>
create ( Scope scope, Operand <T> input, Long groupSize, Long groupKey, Long instanceKey, Shape shape, Options ... options)
Metodo Factory per creare una classe che esegue il wrapping di una nuova operazione CollectiveGather.
Uscita <T>
dati ()
static CollectiveGather.Options
timeoutSeconds (Float timeoutSeconds)

Metodi ereditati

Metodi pubblici

output pubblico <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 CollectiveGather.Options communicationHint (String communicationHint)

public static CollectiveGather <T> create ( Scope scope, Operand <T> input, Long groupSize, Long groupKey, Long instanceKey, Shape shape, Options ... options)

Metodo Factory per creare una classe che esegue il wrapping di una nuova operazione CollectiveGather.

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

output pubblico <T> dati ()

public static CollectiveGather.Options timeoutSeconds (Float timeoutSeconds)