InfeedEnqueue

public final class InfeedEnqueue

Un'operazione che alimenta un singolo valore di tensore nel calcolo.

Classi annidate

classe InfeedEnqueue.Options Attributi facoltativi per InfeedEnqueue

Metodi pubblici

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

Metodi ereditati

Metodi pubblici

public static InfeedEnqueue create ( ambito ambito, operando <T> input, opzioni ... opzioni)

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

Parametri
scopo ambito attuale
ingresso Un tensore che verrà fornito utilizzando il meccanismo di alimentazione.
opzioni trasporta valori di attributi opzionali
ritorna
  • una nuova istanza di InfeedEnqueue

public static InfeedEnqueue.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 InfeedEnqueue.Options statico pubblico ( layout List <Long>)

Parametri
disposizione Un vettore che contiene il layout richiesto in sequenza da minore a maggiore. Se viene passato un attributo di layout, ma i suoi valori sono tutti -1, il layout verrà calcolato dall'operazione di alimentazione.

public static InfeedEnqueue.Options shape ( Shape shape)

Parametri
forma La forma del tensore.