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> | |
Iteratore < Operando <T>> | iteratore () |
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