CrossReplicaSum

classe finale pubblica CrossReplicaSum

Un'operazione per sommare gli input tra le istanze TPU replicate.

Ogni istanza fornisce il proprio input.

Ad esempio, supponiamo che ci siano 8 istanze TPU: "[A, B, C, D, E, F, G, H]". Passando 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 le uscite: `[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] ".

Metodi pubblici

Uscita <T>
asOutput ()
Restituisce la maniglia simbolica di un tensore.
static <T extends Number> CrossReplicaSum <T>
create ( ambito ambito, operando <T> input, operando <Integer> groupAssignment)
Metodo Factory per creare una classe che esegue il wrapping di una nuova operazione CrossReplicaSum.
Uscita <T>
uscita ()
La somma di tutti gli ingressi distribuiti.

Metodi ereditati

Metodi pubblici

output pubblico <T> asOutput ()

Restituisce la maniglia simbolica di un 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 ( Scope scope, Operand <T> input, Operand <Integer> groupAssignment)

Metodo Factory per creare una classe che esegue il wrapping di 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 nel iesimo sottogruppo.
ritorna
  • una nuova istanza di CrossReplicaSum

output pubblico <T> output ()

La somma di tutti gli ingressi distribuiti.