PartitionedOutput public final class
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 particionadas | Los atributos opcionales para PartitionedOutput |
Constantes
Cuerda | OP_NAME | El nombre de esta operación, como lo conoce el motor central de TensorFlow |
Métodos públicos
estática <T se extiende Ttype > PartitionedOutput <T> | crear ( Alcance alcance, operando <T> entradas, numSplits largas, Opciones ... Opciones) Método de fábrica para crear una clase que envuelva una nueva operación PartitionedOutput. |
Iterador < operando <T >> | iterador () |
Lista < salida <T >> | salida () Una lista de entradas particionadas que deben tener la misma forma. |
estáticas PartitionedOutput.Options | partitionDim (Long partitionDim) |
Métodos heredados
Constantes
OP_NAME pública final static String
El nombre de esta operación, como lo conoce el motor central de TensorFlow
Valor constante: "TPUPartitionedOutput"
Métodos públicos
public static PartitionedOutput <T> crear ( Alcance alcance, operando <T> entradas, numSplits largas, Opciones ... Opciones)
Método de fábrica para crear una clase que envuelva una nueva operación PartitionedOutput.
Parámetros
alcance | alcance actual |
---|---|
entradas | Un tensor que representa la forma completa de los tensores divididos. |
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.
public static PartitionedOutput.Options partitionDim (Long partitionDim)
Parámetros
particiónDim | Un número entero describe qué dimensión está dividida. |
---|