TPUReplicatedOutput

TPUReplicatedOutput public final class

Conecta N salidas de un cálculo de TPU replicado de N vías.

Esta operación contiene una salida replicada de un subgrafo de cálculo `tpu.replicate ()`. Cada salida replicada tiene la misma forma y tipo junto con la entrada.

Por ejemplo:

%computation = "tf.Computation"()
 %replicated_output:2 = "tf.TPUReplicatedOutput"(%computation)
 
El cálculo anterior tiene una salida de replicado de dos réplicas.

Métodos públicos

estática <T> TPUReplicatedOutput <T>
crear ( Alcance alcance, operando <T> entrada, Long numReplicas)
Método de fábrica para crear una clase que envuelva una nueva operación TPUReplicatedOutput.
Iterador < operando <T >>
Lista < salida <T >>
salidas ()

Métodos heredados

Métodos públicos

public static TPUReplicatedOutput <T> crear ( Alcance alcance, operando <T> de entrada, Long numReplicas)

Método de fábrica para crear una clase que envuelva una nueva operación TPUReplicatedOutput.

Parámetros
alcance alcance actual
Devoluciones
  • una nueva instancia de TPUReplicatedOutput

pública Iterator < operando <T >> iterador ()

Lista pública < salida <T >> salidas ()