TPUReplicatedOutput

clase final pública TPUReplicatedOutput

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 replicada de dos réplicas.

Métodos públicos

estática <T> TPUReplicatedOutput <T>
crear ( alcance del alcance, entrada del operando <T>, numReplicas largas)
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

TPUReplicatedOutput <T> estático público create ( Scope scope, Operand <T> 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

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

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