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 subgráfico 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ático <T> TPUReplicatedOutput <T>
crear ( ámbito de ámbito, 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

public static TPUReplicatedOutput <T> create ( Scope scope, Operand <T> input, Long numReplicas)

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 ()

public List< Salida <T>> salidas ()