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> | |
Iterador< Operando <T>> | iterador () |
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