clase final pública TPUPartitionedOutput
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 | Atributos opcionales para TPUPartitionedOutput |
Métodos públicos
estática <T> TPUPartitionedOutput <T> | crear ( alcance de alcance, entradas de operando <T>, long numSplits, opciones ... opciones) Método de fábrica para crear una clase que envuelva una nueva operación TPUPartitionedOutput. |
Iterador < Operando <T>> | iterador () |
Lista < Salida <T>> | salida () Una lista de entradas particionadas que deben tener la misma forma. |
estático TPUPartitionedOutput.Options | particiónDim ( ParticiónDim larga) |
Métodos heredados
Métodos públicos
public static TPUPartitionedOutput <T> crear ( alcance de alcance, entradas de operando <T>, long numSplits, 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
lista pública < Salida <T>> salida ()
Una lista de entradas particionadas que deben tener la misma forma.
TPUPartitionedOutput.Options estática pública partitionDim ( PartitionDim larga)
Parámetros
particiónDim | Un número entero describe qué dimensión está dividida. |
---|