Advertencia: esta API está obsoleta y se eliminará en una versión futura de TensorFlow una vez que el reemplazo sea ​​estable.

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]`.

Métodos públicos

Salida <T>
asOutput ()
Devuelve el identificador simbólico de un tensor.
estática <T> CollectivePermute <T>
crear ( Alcance alcance, operando <T> de entrada, operando <Integer> 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

Métodos públicos

pública de salida <T> asOutput ()

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.

public static CollectivePermute <T> crear ( Alcance alcance, operando <T> de entrada, operando <Integer> 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.