CrossReplicaSum

classe final pública CrossReplicaSum

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]`.

Métodos Públicos

Saída <T>
asOutput ()
Retorna o identificador simbólico de um tensor.
estático <T estende número> CrossReplicaSum <T>
create (escopo do escopo , entrada Operando <T>, Operando <Integer> 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

Métodos Públicos

Saída pública <T> asOutput ()

Retorna o identificador simbólico de um 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 Operando <T>, Operando <Integer> 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

Saída pública <T> saída ()

A soma de todas as entradas distribuídas.