classe final pública PartitionedOutput
Uma operação que demultiplexa um tensor a ser fragmentado pelo XLA em uma lista de particionados
saídas fora do cálculo XLA.
Classes aninhadas
aula | PartitionedOutput.Options | Atributos opcionais para PartitionedOutput |
Constantes
Corda | OP_NAME | O nome desta operação, conforme conhecido pelo mecanismo principal do TensorFlow |
Métodos Públicos
estático <T estende TType > PartitionedOutput <T> | create (escopo do escopo , entradas do operando <T>, numSplits longos, opções... opções) Método de fábrica para criar uma classe que envolve uma nova operação PartitionedOutput. |
Iterador< Operando <T>> | iterador () |
Lista< Saída <T>> | saída () Uma lista de entradas particionadas que devem ter o mesmo formato. |
PartitionedOutput.Options estático | partiçãoDim (partição longaDim) |
Métodos herdados
Constantes
String final estática pública OP_NAME
O nome desta operação, conforme conhecido pelo mecanismo principal do TensorFlow
Valor constante: "TPUPartitionedOutput"
Métodos Públicos
public static PartitionedOutput <T> create (escopo do escopo , entradas do operando <T>, numSplits longos, opções... opções)
Método de fábrica para criar uma classe que envolve uma nova operação PartitionedOutput.
Parâmetros
escopo | 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 PartitionedOutput
Lista pública< Saída <T>> saída ()
Uma lista de entradas particionadas que devem ter o mesmo formato.
público estático PartitionedOutput.Options partiçãoDim (partição longaDim)
Parâmetros
partiçãoDim | Um número inteiro descreve qual dimensão é particionada. |
---|