classe final pública TPUPartitionedOutput
Um op que demultiplexa um tensor a ser fragmentado pelo XLA para uma lista de partições
saídas fora da computação XLA.
Classes aninhadas
| aula | TPUPartitionedOutput.Options | Atributos opcionais para TPUPartitionedOutput | |
Métodos públicos
| estático <T> TPUPartitionedOutput <T> | |
| Iterador< Operando <T>> | iterador () |
| Lista< Saída <T>> | saída () Uma lista de entradas particionadas que devem ter a mesma forma. |
| estático TPUPartitionedOutput.Options | partitionDim (Partição longa Dim) |
Métodos Herdados
Métodos públicos
public static TPUPartitionedOutput <T> create ( Escopo do escopo, entradas Operando <T>, Long numSplits, Opções... 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 de tensores particionados. |
| opções | carrega valores de atributos opcionais |
Devoluções
- uma nova instância de TPUPartitionedOutput
public static TPUPartitionedOutput.Options partitionDim (Long partitionDim)
Parâmetros
| partitionDim | Um número inteiro descreve qual dimensão é particionada. |
|---|