InfeedEnqueueTuple

classe finale pubblica InfeedEnqueueTuple

Inserisce più valori tensoriali nel calcolo come una tupla XLA.

Classi nidificate

classe InfeedEnqueueTuple.Options Attributi facoltativi per InfeedEnqueueTuple

Metodi pubblici

InfeedEnqueueTuple statico
create (ambito Scope , input Iterable< Operand <?>>, List< Shape > forme, Opzioni... opzioni)
Metodo factory per creare una classe che racchiude una nuova operazione InfeedEnqueueTuple.
statico InfeedEnqueueTuple.Options
dispositivoOrdinale (Dispositivo lungoOrdinale)
statico InfeedEnqueueTuple.Options
layout (layout Elenco<lungo>)

Metodi ereditati

Metodi pubblici

public static InfeedEnqueueTuple create ( ambito ambito , input Iterable< Operand <?>>, List< Shape > forme, Opzioni... opzioni)

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

Parametri
scopo ambito attuale
input Un elenco di tensori che verranno forniti utilizzando il meccanismo di alimentazione.
forme Le forme di ciascun tensore in "input".
opzioni trasporta valori di attributi opzionali
ritorna
  • una nuova istanza di InfeedEnqueueTuple

public static InfeedEnqueueTuple.Options deviceOrdinal (Long deviceOrdinal)

Parametri
dispositivoOrdinale Il dispositivo TPU da utilizzare. Dovrebbe essere -1 quando l'Operazione è in esecuzione su un dispositivo TPU e >= 0 quando l'Operazione è in esecuzione sul dispositivo CPU.

layout InfeedEnqueueTuple.Options statici pubblici (layout List<Long>)

Parametri
layout Un vettore che contiene il layout richiesto in sequenza da minore a maggiore per tutte le forme della tupla, nell'ordine in cui le forme appaiono nell'input "forme". Gli elementi di layout per una forma secondaria possono essere impostati su -1, nel qual caso il layout corrispondente verrà calcolato dall'operazione di alimentazione.