TPUReplicatedOutput

classe final pública TPUReplicatedOutput

Conecta N saídas de uma computação TPU replicada de N vias.

Esta operação contém uma saída replicada de um subgráfico de computação `tpu.replicate()`. Cada saída replicada tem o mesmo formato e tipo ao lado da entrada.

Por exemplo:

%computation = "tf.Computation"()
 %replicated_output:2 = "tf.TPUReplicatedOutput"(%computation)
 
O cálculo acima tem uma saída replicada de duas réplicas.

Métodos Públicos

estático <T> TPUReplicatedOutput <T>
create (escopo do escopo , entrada do operando <T>, Long numReplicas)
Método de fábrica para criar uma classe que agrupa 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 , entrada do operando <T>, Long numReplicas)

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

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

iterador público< Operando <T>> iterador ()

Lista pública< Saída <T>> saídas ()