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> | |
| 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