classe final pública TPUPartitionedOutputV2
Uma operação que demultiplexa um tensor para ser fragmentado por XLA em uma lista de
saídas fora da computação XLA. Suporta fragmentação ND.
Métodos públicos
static <T> TPUPartitionedOutputV2 <T> | create ( Escopo do escopo , entradas do Operando <T>, Long numSplits, List<Long> partitionDims) Método de fábrica para criar uma classe envolvendo uma nova operação TPUPartitionedOutputV2. |
Iterador < Operando <T>> | iterador () |
Lista< Saída <T>> | saída () Uma lista de saídas particionadas que têm a mesma forma. |
Métodos Herdados
Métodos públicos
public static TPUPartitionedOutputV2 <T> create ( Scope scope, Operand <T> inputs, Long numSplits, List<Long> partitionDims)
Método de fábrica para criar uma classe envolvendo uma nova operação TPUPartitionedOutputV2.
Parâmetros
escopo | escopo atual |
---|---|
entradas | Um tensor que representa a forma completa dos tensores particionados. |
partiçãoDims | Uma lista de inteiros descrevendo como cada dimensão é particionada. O vazio indica que as entradas são replicadas. |
devoluções
- uma nova instância de TPUPartitionedOutputV2