PartitionedOutput

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

pública Iterator < operando <T >> iterador ()

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.