TPUPartitionedOutputV2

clase final pública TPUPartitionedOutputV2

Una operación que demultiplexa 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 (alcance alcance , entradas de operando <T>, numSplits largos, lista <Long> particionDims)
Método de fábrica para crear una clase que envuelve una nueva operación TPUPartitionedOutputV2.
Iterador< Operando <T>>
Lista< Salida <T>>
producción ()
Una lista de salidas particionadas que tienen la misma forma.

Métodos heredados

Métodos públicos

creación pública estática de TPUPartitionedOutputV2 <T> ( alcance , entradas de operando <T>, numSplits largos, lista <Long> particionesDims)

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