CollectivePermute

CollectivePermute classe final pública

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>
criar ( Scope escopo, Operando <T> entrada, Operando < TInt32 > sourceTargetPairs)
Método de fábrica para criar uma classe envolvendo uma nova operação CollectivePermute.
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

Valor constante: "CollectivePermute"

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

pública Output <T> de saída ()

A entrada permutada.