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