TPUReplicatedOutput

classe final pública TPUReplicatedOutput

Conecta N saídas de uma computação TPU replicada em N-way.

Esta operação contém uma saída replicada de um subgrafo de computação `tpu.replicate()`. Cada saída replicada tem a mesma forma e tipo ao lado da entrada.

Por exemplo:

%computation = "tf.Computation"()
 %replicated_output:2 = "tf.TPUReplicatedOutput"(%computation)
 
A computação acima tem uma saída replicada de duas réplicas.

Métodos públicos

estático <T> TPUReplicatedOutput <T>
create ( Escopo do escopo, entrada Operando <T>, Long numReplicas)
Método de fábrica para criar uma classe que envolve uma nova operação TPUReplicatedOutput.
Iterador< Operando <T>>
Lista< Saída <T>>
saídas ()

Métodos Herdados

Métodos públicos

public static TPUReplicatedOutput <T> create ( Escopo do escopo, Operando <T> input, Long numReplicas)

Método de fábrica para criar uma classe que envolve uma nova operação TPUReplicatedOutput.

Parâmetros
alcance escopo atual
Devoluções
  • uma nova instância de TPUReplicatedOutput

public Iterator< Operando <T>> iterador ()

public List< Output <T>> outputs ()