CollectivePermute

clase final pública CollectivePermute

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

Cada instancia proporciona su propia entrada.

Por ejemplo, supongamos que hay 4 instancias de TPU: `[A, B, C, D]`. Al pasar source_target_pairs=`[[0,1],[1,2],[2,3],[3,0]]` se obtienen las salidas: `[D, A, B, C]`.

Métodos públicos

Salida <T>
como salida ()
Devuelve el identificador simbólico de un tensor.
estático <T> CollectivePermute <T>
crear (alcance alcance , entrada Operando <T>, Operando <Integer> sourceTargetPairs)
Método de fábrica para crear una clase que envuelve una nueva operación CollectivePermute.
Salida <T>
producción ()
La entrada permutada.

Métodos heredados

Métodos públicos

Salida pública <T> como Salida ()

Devuelve el identificador simbólico de un 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 CollectivePermute <T> (alcance del alcance , entrada del operando <T>, operando <Integer> sourceTargetPairs)

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

Parámetros
alcance alcance actual
aporte La entrada local que se va a permutar. Actualmente sólo admite float y bfloat16.
fuenteDestinoPairs Un tensor con forma [num_pairs, 2].
Devoluciones
  • una nueva instancia de CollectivePermute

Salida pública <T> salida ()

La entrada permutada.