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