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