clase final pública TPUPartitionedOutput
Una operación que desmultiplexa un tensor para que XLA lo fragmente en una lista de particiones
salidas fuera del cálculo XLA.
Clases anidadas
clase | TPUPartitionedOutput.Options | Atributos opcionales para TPUPartitionedOutput |
Métodos públicos
estático <T> TPUPartitionedOutput <T> | crear ( ámbito de ámbito, entradas de operando <T>, Long numSplits, opciones... opciones) Método de fábrica para crear una clase que envuelve una nueva operación TPUPartitionedOutput. |
Iterador< Operando <T>> | iterador () |
Lista< Salida <T>> | salida () Una lista de entradas particionadas que deben tener la misma forma. |
TPU PartitionedOutput.Options estático | particiónDim (Larga particiónDim) |
Métodos Heredados
Métodos públicos
public static TPUPartitionedOutput <T> crear ( Ámbito de alcance, Operando <T> entradas, Long numSplits, Opciones... opciones)
Método de fábrica para crear una clase que envuelve una nueva operación TPUPartitionedOutput.
Parámetros
alcance | alcance actual |
---|---|
entradas | Un tensor que representa la forma completa de los tensores particionados. |
opciones | lleva valores de atributos opcionales |
Devoluciones
- una nueva instancia de TPUPartitionedOutput
lista pública< salida <T>> salida ()
Una lista de entradas particionadas que deben tener la misma forma.
público estático TPUPartitionedOutput.Options particiónDim (partición largaDim)
Parámetros
particiónDim | Un número entero describe qué dimensión se divide. |
---|