Uma operação para somar entradas em instâncias de TPU replicadas.
Cada instância fornece sua própria entrada.
Por exemplo, suponha que haja 8 instâncias de TPU: `[A, B, C, D, E, F, G, H]`. Passar group_assignment=`[[0,2,4,6],[1,3,5,7]]` define `A, C, E, G` como grupo 0 e `B, D, F, H` como grupo 1. Assim, obtemos as saídas: `[A+C+E+G, B+D+F+H, A+C+E+G, B+D+F+H, A+C+E+ G, B+D+F+H, A+C+E+G, B+D+F+H]`.
Constantes
| Corda | OP_NAME | O nome desta operação, conforme conhecido pelo mecanismo principal do TensorFlow |
Métodos Públicos
| Saída <T> | asOutput () Retorna o identificador simbólico do tensor. |
| estático <T estende TNumber > CrossReplicaSum <T> | create (escopo do escopo , entrada do operando <T>, operando <TInt32> groupAssignment) Método de fábrica para criar uma classe que envolve uma nova operação CrossReplicaSum. |
| Saída <T> | saída () A soma de todas as entradas distribuídas. |
Métodos herdados
Constantes
String final estática pública OP_NAME
O nome desta operação, conforme conhecido pelo mecanismo principal do TensorFlow
Métodos Públicos
Saída pública <T> asOutput ()
Retorna o identificador simbólico do tensor.
As entradas para operações do TensorFlow são saídas de outra operação do TensorFlow. Este método é usado para obter um identificador simbólico que representa o cálculo da entrada.
public static CrossReplicaSum <T> create ( escopo do escopo , entrada do operando <T>, operando < TInt32 > groupAssignment)
Método de fábrica para criar uma classe que envolve uma nova operação CrossReplicaSum.
Parâmetros
| escopo | escopo atual |
|---|---|
| entrada | A entrada local para a soma. |
| atribuição de grupo | Um tensor int32 com forma [num_groups, num_replicas_per_group]. `group_assignment[i]` representa os IDs de réplica no i-ésimo subgrupo. |
Devoluções
- uma nova instância de CrossReplicaSum