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. |
---|