TPUPartitionedOutputV2

clase final pública TPUPartitionedOutputV2

Una operación que desmultiplexa un tensor para que XLA lo fragmente en una lista de particiones

salidas fuera del cálculo XLA. Admite fragmentación ND.

Métodos públicos

estático <T> TPUPartitionedOutputV2 <T>
crear (Ámbito de alcance , Operando <T> entradas, Long numSplits, List<Long> particionesDims)
Método de fábrica para crear una clase que envuelve una nueva operación TPUPartitionedOutputV2.
Iterador< Operando <T>>
Lista< Salida <T>>
salida ()
Una lista de salidas particionadas que tienen la misma forma.

Métodos Heredados

Métodos públicos

public static TPUPartitionedOutputV2 <T> create ( Scope scope, Operando <T> entradas, Long numSplits, List<Long> particionDims)

Método de fábrica para crear una clase que envuelve una nueva operación TPUPartitionedOutputV2.

Parámetros
alcance alcance actual
entradas Un tensor que representa la forma completa de los tensores particionados.
particiónDims Una lista de números enteros que describen cómo se divide cada dimensión. El vacío indica que las entradas se replican.
Devoluciones
  • una nueva instancia de TPUPartitionedOutputV2

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

lista pública< salida <T>> salida ()

Una lista de salidas particionadas que tienen la misma forma.