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