TPUPartitionedOutput

clase final pública TPUPartitionedOutput

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

salidas fuera del cálculo XLA.

Clases anidadas

clase TPUPartitionedOutput.Options Atributos opcionales para TPUPartitionedOutput

Métodos públicos

estático <T> TPUPartitionedOutput <T>
crear ( ámbito de ámbito, entradas de operando <T>, Long numSplits, opciones... opciones)
Método de fábrica para crear una clase que envuelve una nueva operación TPUPartitionedOutput.
Iterador< Operando <T>>
Lista< Salida <T>>
salida ()
Una lista de entradas particionadas que deben tener la misma forma.
TPU PartitionedOutput.Options estático
particiónDim (Larga particiónDim)

Métodos Heredados

Métodos públicos

public static TPUPartitionedOutput <T> crear ( Ámbito de alcance, Operando <T> entradas, Long numSplits, Opciones... opciones)

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

Parámetros
alcance alcance actual
entradas Un tensor que representa la forma completa de los tensores particionados.
opciones lleva valores de atributos opcionales
Devoluciones
  • una nueva instancia de TPUPartitionedOutput

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

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

Una lista de entradas particionadas que deben tener la misma forma.

público estático TPUPartitionedOutput.Options particiónDim (partición largaDim)

Parámetros
particiónDim Un número entero describe qué dimensión se divide.