TPUPartitionedOutputV2

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

iterador público< Operando <T>> iterador ()

Lista pública< Saída <T>> saída ()

Uma lista de saídas particionadas que possuem o mesmo formato.