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

Constantes

Cuerda OP_NAME El nombre de esta operación, como lo conoce el motor central de TensorFlow

Métodos públicos

Salida <T>
asOutput ()
Devuelve el identificador simbólico del tensor.
estática <T se extiende TNumber > CrossReplicaSum <T>
crear ( Alcance alcance, operando <T> de entrada, operando < TInt32 > 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

org.tensorflow.op.RawOp
booleano final
es igual a (Object obj)
int final
hashCode ()
Operación
op ()
Devuelva esta unidad de cálculo como una sola Operation .
cadena final
toString ()
booleano
equals (Object arg0)
Clase final <?>
getClass ()
En t
hashCode ()
vacío final
notificar ()
vacío final
notifyAll ()
Cuerda
toString ()
vacío final
espera (tiempo arg0, arg1 int)
vacío final
espera (arg0 largo)
vacío final
wait ()
org.tensorflow.op.Op
abstracta ExecutionEnvironment
env ()
Devuelve el entorno de ejecución en el que se creó esta operación.
abstracta Operación
op ()
Devuelva esta unidad de cálculo como una sola Operation .
org.tensorflow.Operand
abstracta de salida <T>
asOutput ()
Devuelve el identificador simbólico del tensor.
T abstracto
asTensor ()
Devuelve el tensor en este operando.
abstracta Forma
forma ()
Devuelve el (posiblemente parcialmente conocida) forma del tensor se hace referencia por la Output de esta operando.
clase abstracta <T>
escriba ()
Devuelve el tipo de tensor de este operando
org.tensorflow.ndarray.Shaped
int abstracto
rango ()
abstracta Forma
forma ()
abstracto largo
tamaño ()
Calcula y devuelve el tamaño total de este contenedor, en número de valores.

Constantes

OP_NAME pública final static String

El nombre de esta operación, como lo conoce el motor central de TensorFlow

Valor constante: "CrossReplicaSum"

Métodos públicos

pública de salida <T> asOutput ()

Devuelve el identificador simbólico del 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 < TInt32 > 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.