TPUPartitionedOutput

classe finale pubblica TPUPartitionedOutput

Un'operazione che demultiplexa un tensore affinché venga suddiviso da XLA in un elenco di partizioni

output al di fuori del calcolo XLA.

Classi nidificate

classe TPUPartitionedOutput.Options Attributi facoltativi per TPUPartitionedOutput

Metodi pubblici

<T> statico TPUPartitionedOutput <T>
create (ambito ambito , input operando <T>, numSplit lunghi, opzioni... opzioni)
Metodo factory per creare una classe che racchiude una nuova operazione TPUPartitionedOutput.
Iteratore< Operando <T>>
Elenco< Uscita <T>>
produzione ()
Un elenco di input partizionati che devono avere la stessa forma.
TPUPartitionedOutput.Options statico
partizioneDim (partizioneDim lunga)

Metodi ereditati

Metodi pubblici

public static TPUPartitionedOutput <T> create (ambito ambito , input operando <T>, numSplits lunghi, opzioni... opzioni)

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

Parametri
scopo ambito attuale
input Un tensore che rappresenta la forma completa dei tensori partizionati.
opzioni trasporta valori di attributi opzionali
ritorna
  • una nuova istanza di TPUPartitionedOutput

iteratore pubblico< operando <T>> iteratore ()

elenco pubblico< Uscita <T>> uscita ()

Un elenco di input partizionati che devono avere la stessa forma.

public static TPUPartitionedOutput.Options partizioneDim (partizione lunga)

Parametri
partizioneDim Un numero intero descrive quale dimensione è partizionata.