classe final pública CollectivePermute
Uma operação para permutar tensores em instâncias de TPU replicadas.
Cada instância fornece sua própria entrada.
Por exemplo, suponha que existam 4 instâncias de TPU: `[A, B, C, D]`. Passar source_target_pairs=`[[0,1],[1,2],[2,3],[3,0]]` obtém as saídas: `[D, A, B, C]`.
Constantes
| Corda | OP_NAME | O nome desta operação, conforme conhecido pelo mecanismo principal do TensorFlow |
Métodos Públicos
| Saída <T> | asOutput () Retorna o identificador simbólico do tensor. |
| estático <T estende TType > CollectivePermute <T> | create ( escopo do escopo , entrada do operando <T>, operando < TInt32 > sourceTargetPairs) Método de fábrica para criar uma classe que envolve uma nova operação CollectPermute. |
| Saída <T> | saída () A entrada permutada. |
Métodos herdados
Constantes
String final estática pública OP_NAME
O nome desta operação, conforme conhecido pelo mecanismo principal do TensorFlow
Valor constante: "CollectivePermute"
Métodos Públicos
Saída pública <T> asOutput ()
Retorna o identificador simbólico do tensor.
As entradas para 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 ( Escopo do escopo , Operando <T> entrada, Operando < TInt32 > sourceTargetPairs)
Método de fábrica para criar uma classe que envolve uma nova operação CollectPermute.
Parâmetros
| escopo | escopo atual |
|---|---|
| entrada | A entrada local a ser permutada. Atualmente suporta apenas float e bfloat16. |
| fonteTargetPairs | Um tensor com forma [num_pairs, 2]. |
Devoluções
- uma nova instância de CollectivePermute