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. |
static <T> CollectivePermute <T> | create ( alcance del alcance, entrada del operando <T>, operando <entero> 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
Salida pública <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> create ( Scope scope, Operand <T> input, Operand <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