TPUPartitionedOutput

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 >>
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

pública Iterator < Operando <T >> iterator ()

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.