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 (ámbito de alcance , entrada de operando <T>, número largo de réplicas)
Método de fábrica para crear una clase que envuelve una nueva operación TPUReplicatedOutput.
Iterador< Operando <T>>
Lista< Salida <T>>
salidas ()

Métodos heredados

Métodos públicos

creación pública estática de TPUReplicatedOutput <T> (alcance del alcance , entrada del operando <T>, número largo de réplicas)

Método de fábrica para crear una clase que envuelve 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 ()