Um Op para permutar tensores em instâncias replicadas de TPU.
Cada instância fornece sua própria entrada.
Por exemplo, suponha que haja 4 instâncias de TPU: `[A, B, C, D]`. Passando source_target_pairs = `[[0,1], [1,2], [2,3], [3,0]]` obtém-se as saídas: `[D, A, B, C]`.
Constantes
Fragmento | OP_NAME | O nome desta operação, conforme conhecido pelo motor principal TensorFlow |
Métodos Públicos
Output <T> | asOutput () Retorna o identificador simbólico do tensor. |
estática <T estende TType > CollectivePermute <T> | |
Output <T> | saída () A entrada permutada. |
Métodos herdados
Constantes
nome_op final String public static
O nome desta operação, conforme conhecido pelo motor principal TensorFlow
Métodos Públicos
pública Output <T> asOutput ()
Retorna o identificador simbólico do tensor.
As entradas para as operações do TensorFlow são saídas de outra operação do TensorFlow. Este método é usado para obter um identificador simbólico que representa o cálculo da entrada.
public static CollectivePermute <T> create ( Scope escopo, Operando <T> entrada, Operando < TInt32 > sourceTargetPairs)
Método de fábrica para criar uma classe envolvendo uma nova operação CollectivePermute.
Parâmetros
alcance | escopo atual |
---|---|
entrada | A entrada local a ser permutada. Atualmente suporta apenas float e bfloat16. |
sourceTargetPairs | Um tensor com forma [num_pares, 2]. |
Devoluções
- uma nova instância de CollectivePermute