CrossReplicaSum

CrossReplicaSum classe final pública

Um Op 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]`. Passando 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

Output <T>
asOutput ()
Retorna o identificador simbólico de um tensor.
estática <T estende Number> CrossReplicaSum <T>
criar ( Scope escopo, Operando <T> entrada, Operando <Integer> groupAssignment)
Método de fábrica para criar uma classe envolvendo uma nova operação CrossReplicaSum.
Output <T>
saída ()
A soma de todas as entradas distribuídas.

Métodos herdados

Métodos Públicos

pública Output <T> asOutput ()

Retorna o identificador simbólico de um tensor.

As entradas para as 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 ( Scope escopo, Operando <T> entrada, Operando <Integer> groupAssignment)

Método de fábrica para criar uma classe envolvendo uma nova operação CrossReplicaSum.

Parâmetros
alcance escopo atual
entrada A entrada local para a soma.
groupAssignment Um tensor int32 com forma [num_groups, num_replicas_per_group]. `group_assignment [i]` representa os ids de réplica no iº subgrupo.
Devoluções
  • uma nova instância de CrossReplicaSum

pública Output <T> de saída ()

A soma de todas as entradas distribuídas.