AllToAll

clase final pública AllToAll

Una operació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 dada la asignación_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, supongamos 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

Cadena

OP_NOMBRE

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

Métodos públicos

Salida <T>	como salida () Devuelve el identificador simbólico del tensor.
estático <T extiende TType > AllToAll <T>	crear (alcance del alcance , entrada del operando <T>, operación <TInt32> groupAssignment, Long concatDimension, Long splitDimension, Long splitCount) Método de fábrica para crear una clase que envuelve una nueva operación AllToAll.
Salida <T>	producción () El resultado intercambiado.

Métodos heredados

De la clase org.tensorflow.op.RawOp

booleano final	es igual (Objeto obj)
entero final	código hash ()
Operación	operación () Devuelve esta unidad de cálculo como una única Operation .
cadena final	Encadenar ()

De la clase java.lang.Object

booleano	es igual (Objeto arg0)
Clase final<?>	obtenerclase ()
En t	código hash ()
vacío final	notificar ()
vacío final	notificar a todos ()
Cadena	Encadenar ()
vacío final	esperar (arg0 largo, int arg1)
vacío final	espera (largo arg0)
vacío final	esperar ()

Desde la interfaz org.tensorflow.op.Op

entorno de ejecución abstracto	entorno () Devuelve el entorno de ejecución en el que se creó esta operación.
operación abstracta	operación () Devuelve esta unidad de cálculo como una única Operation .

Desde la interfaz org.tensorflow.Operand

Salida abstracta <T>	como salida () Devuelve el identificador simbólico del tensor.
resumen T	como Tensor () Devuelve el tensor en este operando.
forma abstracta	forma () Devuelve la forma (posiblemente parcialmente conocida) del tensor al que hace referencia la Output de este operando.
clase abstracta<T>	tipo () Devuelve el tipo de tensor de este operando.

Desde la interfaz org.tensorflow.ndarray.Shaped

resumen entero	rango ()
forma abstracta	forma ()
abstracto largo	tamaño () Calcula y devuelve el tamaño total de este contenedor, en número de valores.

Constantes

Cadena final estática pública OP_NAME

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

Valor constante: "Todos a todos"

Métodos públicos

Salida pública <T> como Salida ()

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.

creación pública estática AllToAll <T> (alcance del alcance , entrada del operando <T>, operación <TInt32> groupAssignment, Long concatDimension, Long splitDimension, Long splitCount)

Método de fábrica para crear una clase que envuelve 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 identificadores de réplica en el subgrupo i.
concatDimensión	El número de dimensión que se va a concatenar.
dimensión dividida	El número de dimensión que se va a dividir.
dividirContar	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

Salida pública <T> salida ()

El resultado intercambiado.