InfeedEnqueueTuple.Options

classe statica pubblica InfeedEnqueueTuple.Options

Attributi facoltativi per InfeedEnqueueTuple

Metodi pubblici

InfeedEnqueueTuple.Options
dispositivoOrdinale (Dispositivo lungoOrdinale)
InfeedEnqueueTuple.Options
layout (layout Elenco<lungo>)

Metodi ereditati

Metodi pubblici

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