PartitionedOutput

clase final pública PartitionedOutput

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

salidas fuera del cálculo XLA.

Clases anidadas

clase Opciones de salida particionada Atributos opcionales para PartitionedOutput

Constantes

Cadena OP_NOMBRE El nombre de esta operación, como lo conoce el motor central de TensorFlow.

Métodos públicos

estático <T extiende TType > PartitionedOutput <T>
crear (alcance alcance , entradas de operando <T>, divisiones de números largos, opciones... opciones)
Método de fábrica para crear una clase que envuelve una nueva operación PartitionedOutput.
Iterador< Operando <T>>
Lista< Salida <T>>
producción ()
Una lista de entradas particionadas que deben tener la misma forma.
Opciones de salida particionada estática
particiónDim (partición largaDim)

Métodos heredados

Constantes

Cadena final estática pública OP_NAME

El nombre de esta operación, como lo conoce el motor central de TensorFlow.

Valor constante: "TPUPartitionedOutput"

Métodos públicos

creación de PartitionedOutput <T> estática pública (alcance de alcance , entradas de operando <T>, divisiones de números largos, opciones... opciones)

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

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

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

Lista pública< Salida <T>> salida ()

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

PartitionedOutput.Options estática pública particiónDim (partición largaDim)

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