clase final pública CollectivePermute
Una operación para permutar tensores entre instancias de TPU replicadas.
Cada instancia proporciona su propia entrada.
Por ejemplo, supongamos que hay 4 instancias de TPU: `[A, B, C, D]`. Al pasar source_target_pairs=`[[0,1],[1,2],[2,3],[3,0]]` se obtienen las salidas: `[D, A, B, C]`.
Constantes
| Cadena | OP_NOMBRE | El nombre de esta operación, como lo conoce el motor central de TensorFlow. |
Métodos públicos
| Salida <T> | como salida () Devuelve el identificador simbólico del tensor. |
| estático <T extiende TType > CollectivePermute <T> | crear (alcance alcance , entrada del operando <T>, operando <TInt32> sourceTargetPairs) Método de fábrica para crear una clase que envuelve una nueva operación CollectivePermute. |
| Salida <T> | producción () La entrada permutada. |
Métodos heredados
Constantes
Cadena final estática pública OP_NAME
El nombre de esta operación, como lo conoce el motor central de TensorFlow.
Valor constante: "CollectivePermute"
Métodos públicos
Salida pública <T> como Salida ()
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.
creación pública estática de CollectivePermute <T> (alcance del alcance , entrada del operando <T>, operando <TInt32> 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 sólo admite float y bfloat16. |
| fuenteDestinoPairs | Un tensor con forma [num_pairs, 2]. |
Devoluciones
- una nueva instancia de CollectivePermute