TPUReplicatedOutput

public final class TPUReplicatedOutput

Collega N uscite da un calcolo TPU replicato a N vie.

Questa operazione contiene un output replicato da un sottografo di calcolo `tpu.replicate ()`. Ogni output replicato ha la stessa forma e tipo accanto all'input.

Ad esempio:

%computation = "tf.Computation"()
 %replicated_output:2 = "tf.TPUReplicatedOutput"(%computation)
 
Il calcolo precedente ha un output replicato di due repliche.

Metodi pubblici

static <T> TPUReplicatedOutput <T>
create ( ambito ambito, operando <T> input, long numReplicas)
Metodo Factory per creare una classe che avvolge una nuova operazione TPUReplicatedOutput.
Iteratore < Operando <T>>
Elenco < Uscita <T>>
uscite ()

Metodi ereditati

Metodi pubblici

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

Metodo Factory per creare una classe che esegue il wrapping di una nuova operazione TPUReplicatedOutput.

Parametri
scopo ambito attuale
ritorna
  • una nuova istanza di TPUReplicatedOutput

iteratore pubblico < Operando <T>> iteratore ()

public List < Output <T>> output ()