TPUPartitionedOutput

public final class TPUPartitionedOutput

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

iteratore pubblico < Operando <T>> iteratore ()

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.