Una operación que demultiplexa un tensor para que XLA lo fragmente en una lista de particiones
salidas fuera del cálculo XLA. Admite fragmentación ND.
Métodos públicos
estático <T> TPUPartitionedOutputV2 <T> | crear (alcance alcance , entradas de operando <T>, numSplits largos, lista <Long> particionDims) Método de fábrica para crear una clase que envuelve una nueva operación TPUPartitionedOutputV2. |
Iterador< Operando <T>> | iterador () |
Lista< Salida <T>> | producción () Una lista de salidas particionadas que tienen la misma forma. |
Métodos heredados
Métodos públicos
creación pública estática de TPUPartitionedOutputV2 <T> ( alcance , entradas de operando <T>, numSplits largos, lista <Long> particionesDims)
Método de fábrica para crear una clase que envuelve una nueva operación TPUPartitionedOutputV2.
Parámetros
alcance | alcance actual |
---|---|
entradas | Un tensor que representa la forma completa de tensores particionados. |
particiónDims | Una lista de números enteros que describen cómo se divide cada dimensión. El vacío indica que las entradas se replican. |
Devoluciones
- una nueva instancia de TPUPartitionedOutputV2