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 | Opciones de salida particionada de TPU | Atributos opcionales para TPUPartitionedOutput | |
Métodos públicos
| estática <T> TPUparticionadaSalida <T> | crear (alcance alcance , entradas de operando <T>, divisiones de números largos, 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>> | producción () Una lista de entradas particionadas que deben tener la misma forma. |
| Opciones de salida particionada de TPU estática | particiónDim (partición largaDim) |
Métodos heredados
Métodos públicos
creación pública estática de TPUPartitionedOutput <T> (alcance de alcance , entradas de operando <T>, divisiones de números largos, 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 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.
TPUPartitionedOutput.Options estática pública particiónDim (partición largaDim)
Parámetros
| particiónDim | Un número entero describe qué dimensión está dividida. |
|---|