CollectivePermute

CollectivePermute public final class

Una operación para permutar tensores en instancias de TPU replicadas.

Cada instancia proporciona su propia entrada.

Por ejemplo, suponga que hay 4 instancias de TPU: "[A, B, C, D]". Pasando source_target_pairs = `[[0,1], [1,2], [2,3], [3,0]]` obtiene los resultados: `[D, A, B, C]`.

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 > CollectivePermute <T>	crear ( Alcance alcance, operando <T> de entrada, operando < TInt32 > sourceTargetPairs) Método de fábrica para crear una clase que envuelva una nueva operación CollectivePermute.
Salida <T>	salida () La entrada permutada.

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: "CollectivePermute"

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 CollectivePermute <T> crear ( Alcance alcance, operando <T> de entrada, operando < TInt32 > sourceTargetPairs)

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

Parámetros

alcance	alcance actual
aporte	La entrada local que se va a permutar. Actualmente solo admite float y bfloat16.
sourceTargetPairs	Un tensor con forma [num_pairs, 2].

Devoluciones

• una nueva instancia de CollectivePermute

pública de salida <T> salida ()

La entrada permutada.