Advertencia: esta API está obsoleta y se eliminará en una versión futura de TensorFlow una vez que el reemplazo sea ​​estable.

TPUPartitionedOutput

Organiza tus páginas con colecciones Guarda y categoriza el contenido según tus preferencias.
TPUPartitionedOutput 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 TPUPartitionedOutput.Options Los atributos opcionales para TPUPartitionedOutput

Métodos públicos

estática <T> TPUPartitionedOutput <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 TPUPartitionedOutput.
Iterador < operando <T >>
Lista < salida <T >>
salida ()
Una lista de entradas particionadas que deben tener la misma forma.
estáticas TPUPartitionedOutput.Options
partitionDim (Long partitionDim)

Métodos heredados

Métodos públicos

public static TPUPartitionedOutput <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 TPUPartitionedOutput.

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 TPUPartitionedOutput

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 TPUPartitionedOutput.Options partitionDim (Long partitionDim)

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