clase final pública TPUPartitionedOutputV2
Una operación que desmultiplexa 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 (Ámbito de alcance , Operando <T> entradas, Long numSplits, List<Long> particionesDims) Método de fábrica para crear una clase que envuelve una nueva operación TPUPartitionedOutputV2. |
Iterador< Operando <T>> | iterador () |
Lista< Salida <T>> | salida () Una lista de salidas particionadas que tienen la misma forma. |
Métodos Heredados
Métodos públicos
public static TPUPartitionedOutputV2 <T> create ( Scope scope, Operando <T> entradas, Long numSplits, List<Long> particionDims)
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 los 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