Un'operazione che demultipla un tensore che deve essere suddiviso da XLA in un elenco di partizionati
uscite al di fuori del calcolo XLA.
Classi annidate
classe | TPUPartitionedOutput.Options | Attributi facoltativi per TPUPartitionedOutput |
Metodi pubblici
static <T> TPUPartitionedOutput <T> | create ( Scope scope, Operand <T> input, Long numSplits, Options ... options) Metodo Factory per creare una classe che avvolge una nuova operazione TPUPartitionedOutput. |
Iteratore < Operando <T>> | iteratore () |
Elenco < Uscita <T>> | uscita () Un elenco di input partizionati che devono avere la stessa forma. |
static TPUPartitionedOutput.Options | partitionDim (Long partitionDim) |
Metodi ereditati
Metodi pubblici
public static TPUPartitionedOutput <T> create ( Scope scope, Operand <T> input, Long numSplits, Options ... options)
Metodo Factory per creare una classe che esegue il wrapping di una nuova operazione TPUPartitionedOutput.
Parametri
scopo | ambito attuale |
---|---|
ingressi | Un tensore che rappresenta la forma completa dei tensori partizionati. |
opzioni | trasporta valori di attributi opzionali |
ritorna
- una nuova istanza di TPUPartitionedOutput
public List < Output <T>> output ()
Un elenco di input partizionati che devono avere la stessa forma.
public static TPUPartitionedOutput.Options partitionDim (Long partitionDim)
Parametri
partitionDim | Un numero intero descrive quale dimensione è partizionata. |
---|