InfeedEnqueueTuple

public final class InfeedEnqueueTuple

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

Classi annidate

classe InfeedEnqueueTuple.Options Attributi facoltativi per InfeedEnqueueTuple

Metodi pubblici

static InfeedEnqueueTuple
create ( ambito ambito, input < operando <? >> iterabile, forme elenco < forma >, opzioni ... opzioni)
Metodo Factory per creare una classe che esegue il wrapping di una nuova operazione InfeedEnqueueTuple.
static InfeedEnqueueTuple.Options
deviceOrdinal (dispositivo lungoOrdinale)
static InfeedEnqueueTuple.Options
layout (List <Long> layouts)

Metodi ereditati

Metodi pubblici

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

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

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

public static InfeedEnqueueTuple.Options deviceOrdinal (Long deviceOrdinal)

Parametri
deviceOrdinal Il dispositivo TPU da utilizzare. Dovrebbe essere -1 quando Op è in esecuzione su un dispositivo TPU e> = 0 quando Op è in esecuzione sul dispositivo CPU.

layout statico pubblico InfeedEnqueueTuple.Options ( 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 sottoforma possono essere impostati su -1, nel qual caso il layout corrispondente verrà calcolato dall'operazione di alimentazione.