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

CollectivePermute

Organiza tus páginas con colecciones Guarda y categoriza el contenido según tus preferencias.
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.