AllToAll

AllToAll public final class

Una opción para intercambiar datos entre réplicas de TPU.

En cada réplica, la entrada se divide en bloques `split_count` a lo largo de` split_dimension` y se envía a las otras réplicas según la asignación de grupo. Después de recibir `split_count` - 1 bloques de otras réplicas, concatenamos los bloques a lo largo de` concat_dimension` como salida.

Por ejemplo, suponga que hay 2 réplicas de TPU: la réplica 0 recibe la entrada: `[[A, B]]` la réplica 1 recibe la entrada: `[[C, D]]`

group_assignment = `[[0, 1]]` concat_dimension = 0 split_dimension = 1 split_count = 2

salida de la réplica 0: `[[A], [C]]` salida de la réplica 1: `[[B], [D]]`

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 Ttype > AllToAll <T>	crear ( Alcance alcance, operando <T> de entrada, operando < TInt32 > groupAssignment, Long concatDimension, Long splitDimension, Long splitCount) Método de fábrica para crear una clase que envuelva una nueva operación AllToAll.
Salida <T>	salida () El resultado intercambiado.

Métodos heredados

De la clase 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 ()

De la clase java.lang.Object

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 ()

Desde la interfaz 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 .

Desde la interfaz 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

Desde la interfaz 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: "AllToAll"

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 AllToAll <T> crear ( Alcance alcance, operando <T> de entrada, operando < TInt32 > groupAssignment, Long concatDimension, Long splitDimension, Long splitCount)

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

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.
concatDimension	El número de dimensión para concatenar.
splitDimension	El número de dimensión que se dividirá.
splitCount	El número de divisiones, este número debe ser igual al tamaño del subgrupo (group_assignment.get_shape () [1])

Devoluciones

• una nueva instancia de AllToAll

pública de salida <T> salida ()

El resultado intercambiado.