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>> | iterador () |
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
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. |
---|