clase final pública CollectivePermute
Op para permutar tensores en instancias de TPU replicadas.
Cada instancia proporciona su propia entrada.
Por ejemplo, supongamos que hay 4 instancias de TPU: `[A, B, C, D]`. Pasar source_target_pairs=`[[0,1],[1,2],[2,3],[3,0]]` obtiene los resultados: `[D, A, B, C]`.
Métodos públicos
| Salida <T> | como salida () Devuelve el identificador simbólico de un tensor. |
| <T> estático Permutación colectiva <T> | |
| Salida <T> | salida () 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 de 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> create ( Scope scope, Operand <T> input, Operand <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 solo admite float y bfloat16. |
| fuenteObjetivoPares | Un tensor con forma [num_pairs, 2]. |
Devoluciones
- una nueva instancia de CollectivePermute