Un'operazione per sommare gli input tra le istanze TPU replicate.
Ogni istanza fornisce il proprio input.
Ad esempio, supponiamo che siano presenti 8 istanze TPU: "[A, B, C, D, E, F, G, H]". Il passaggio di group_assignment=`[[0,2,4,6],[1,3,5,7]]` imposta "A, C, E, G" come gruppo 0 e "B, D, F, H" come gruppo 1. Quindi otteniamo i risultati: `[A+C+E+G, B+D+F+H, A+C+E+G, B+D+F+H, A+C+E+ SOL, B+D+F+H, A+C+E+G, B+D+F+H]`.
Costanti
| Corda | OP_NAME | Il nome di questa operazione, come noto al motore principale di TensorFlow |
Metodi pubblici
| Uscita <T> | comeuscita () Restituisce l'handle simbolico del tensore. |
| statico <T estende TNumber > CrossReplicaSum <T> | |
| Uscita <T> | produzione () La somma di tutti gli input distribuiti. |
Metodi ereditati
Costanti
Stringa finale statica pubblica OP_NAME
Il nome di questa operazione, come noto al motore principale di TensorFlow
Metodi pubblici
Uscita pubblica <T> asOutput ()
Restituisce l'handle simbolico del tensore.
Gli input per le operazioni TensorFlow sono output di un'altra operazione TensorFlow. Questo metodo viene utilizzato per ottenere un handle simbolico che rappresenta il calcolo dell'input.
public static CrossReplicaSum <T> create ( ambito ambito , operando <T> input, operando < TInt32 > groupAssignment)
Metodo factory per creare una classe che racchiude una nuova operazione CrossReplicaSum.
Parametri
| scopo | ambito attuale |
|---|---|
| ingresso | L'input locale alla somma. |
| groupAssignment | Un tensore int32 con forma [num_groups, num_replicas_per_group]. "group_assignment[i]" rappresenta gli ID di replica nell'i-esimo sottogruppo. |
ritorna
- una nuova istanza di CrossReplicaSum