CrossReplicaSum

CrossReplicaSum public final class

Una operación para sumar entradas en instancias de TPU replicadas.

Cada instancia proporciona su propia entrada.

Por ejemplo, suponga que hay 8 instancias de TPU: "[A, B, C, D, E, F, G, H]". Pasar group_assignment = `[[0,2,4,6], [1,3,5,7]]` establece `A, C, E, G` como grupo 0 y` B, D, F, H` como grupo 1. Así obtenemos las salidas: `[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

Salida <T>
asOutput ()
Devuelve el identificador simbólico de un tensor.
estática <T se extiende Número> CrossReplicaSum <T>
crear ( Alcance alcance, operando <T> de entrada, operando <Integer> groupAssignment)
Método de fábrica para crear una clase que envuelva una nueva operación CrossReplicaSum.
Salida <T>
salida ()
La suma de todas las entradas distribuidas.

Métodos heredados

Métodos públicos

pública de salida <T> asOutput ()

Devuelve el identificador simbólico de un tensor.

Las entradas a las operaciones de TensorFlow son salidas de otra operación de TensorFlow. Este método se utiliza para obtener un identificador simbólico que representa el cálculo de la entrada.

public static CrossReplicaSum <T> crear ( Alcance alcance, operando <T> de entrada, operando <Integer> groupAssignment)

Método de fábrica para crear una clase que envuelva una nueva operación CrossReplicaSum.

Parámetros
alcance alcance actual
aporte La entrada local a la suma.
asignación grupal Un tensor int32 con forma [num_groups, num_replicas_per_group]. `group_assignment [i]` representa los ID de réplica en el i-ésimo subgrupo.
Devoluciones
  • una nueva instancia de CrossReplicaSum

pública de salida <T> salida ()

La suma de todas las entradas distribuidas.