PartitionedOutput

PartitionedOutput classe final pública

Uma operação que demultiplexa um tensor a ser fragmentado por XLA em uma lista de particionados

saídas fora da computação XLA.

Classes aninhadas

classe PartitionedOutput.Options Atributos opcionais para PartitionedOutput

Constantes

Fragmento OP_NAME O nome desta operação, conforme conhecido pelo motor principal TensorFlow

Métodos Públicos

estática <T estende TType > PartitionedOutput <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 PartitionedOutput.
Iterator < Operando <T >>
List < Output <T >>
saída ()
Uma lista de entradas particionadas que devem ter o mesmo formato.
estáticos PartitionedOutput.Options
partitionDim (Long partitionDim)

Métodos herdados

Constantes

nome_op final String public static

O nome desta operação, conforme conhecido pelo motor principal TensorFlow

Valor constante: "TPUPartitionedOutput"

Métodos Públicos

public static PartitionedOutput <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 PartitionedOutput.

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 PartitionedOutput

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 PartitionedOutput.Options partitionDim (Long partitionDim)

Parâmetros
partiçãoDim Um inteiro descreve qual dimensão é particionada.