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

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.