TPUPartitionedOutputV2

classe finale pubblica TPUPartitionedOutputV2

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

output al di fuori del calcolo XLA. Supporta lo sharding ND.

Metodi pubblici

<T> statico TPUPartitionedOutputV2 <T>
create ( ambito ambito , ingressi operando <T>, numSplits lunghi, List<Long> partizioneDims)
Metodo factory per creare una classe che racchiude una nuova operazione TPUPartitionedOutputV2.
Iteratore< Operando <T>>
iteratore ()
Elenco< Uscita <T>>
produzione ()
Un elenco di output partizionati che hanno la stessa forma.

Metodi ereditati

Metodi pubblici

public static TPUPartitionedOutputV2 <T> create ( scope scope, operando <T> input, Long numSplits, List<Long> partitionDims)

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

Parametri
scopo ambito attuale
input Un tensore che rappresenta la forma completa dei tensori partizionati.
partizioneDims Un elenco di numeri interi che descrivono come è partizionata ciascuna dimensione. Il vuoto indica che gli input vengono replicati.
ritorna
  • una nuova istanza di TPUPartitionedOutputV2

iteratore pubblico< operando <T>> iteratore ()

elenco pubblico< Uscita <T>> uscita ()

Un elenco di output partizionati che hanno la stessa forma.