TPUPartitionedOutput classe final pública
Uma operação que demultiplexa um tensor a ser fragmentado por XLA em uma lista de
saídas fora da computação XLA.
Classes aninhadas
| classe | TPUPartitionedOutput.Options | Atributos opcionais para TPUPartitionedOutput | |
Métodos Públicos
| estática <T> TPUPartitionedOutput <T> | criar ( Scope escopo, Operando <T> entradas, numSplits longas, Options ... Opções) Método de fábrica para criar uma classe que envolve uma nova operação TPUPartitionedOutput. |
| Iterator < Operando <T >> | iteração () |
| List < Output <T >> | saída () Uma lista de entradas particionadas que devem ter o mesmo formato. |
| estáticos TPUPartitionedOutput.Options | partitionDim (Long partitionDim) |
Métodos herdados
Métodos Públicos
public static TPUPartitionedOutput <T> create ( Scope escopo, Operando <T> entradas, numSplits longas, Options ... Opções)
Método de fábrica para criar uma classe que envolve uma nova operação TPUPartitionedOutput.
Parâmetros
| alcance | escopo atual |
|---|---|
| entradas | Um tensor que representa a forma completa dos tensores particionados. |
| opções | carrega valores de atributos opcionais |
Devoluções
- uma nova instância de TPUPartitionedOutput
Lista pública < Output <T >> saída ()
Uma lista de entradas particionadas que devem ter o mesmo formato.
public static TPUPartitionedOutput.Options partitionDim (Long partitionDim)
Parâmetros
| partiçãoDim | Um inteiro descreve qual dimensão é particionada. |
|---|