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>> | iteratore () |
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
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. |
---|