classe final pública TPUPartitionedOutputV2
Uma operação que demultiplexa um tensor a ser fragmentado pelo XLA em uma lista de particionados
saídas fora do cálculo XLA. Suporta fragmentação ND.
Métodos Públicos
estático <T> TPUPartitionedOutputV2 <T> | create (escopo do escopo , entradas do operando <T>, NumSplits longos, List<Long> partiçãoDims) Método de fábrica para criar uma classe que agrupa uma nova operação TPUPartitionedOutputV2. |
Iterador< Operando <T>> | iterador () |
Lista< Saída <T>> | saída () Uma lista de saídas particionadas que possuem o mesmo formato. |
Métodos herdados
Métodos Públicos
public static TPUPartitionedOutputV2 <T> create (escopo do escopo , entradas do operando <T>, NumSplits longos, List<Long> partiçãoDims)
Método de fábrica para criar uma classe que agrupa uma nova operação TPUPartitionedOutputV2.
Parâmetros
escopo | escopo atual |
---|---|
entradas | Um tensor que representa a forma completa dos tensores particionados. |
partiçãoDims | Uma lista de números inteiros que descreve como cada dimensão é particionada. O vazio indica que as entradas são replicadas. |
Devoluções
- uma nova instância de TPUPartitionedOutputV2