Aviso: esta API está obsoleta e será removida em uma versão futura do TensorFlow depois que a substituição estiver estável.

CollectivePermute

public final class CollectivePermute

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]`.

Métodos Públicos

Saída <T>
asOutput ()
Retorna o identificador simbólico de um tensor.
static <T> CollectivePermute <T>
criar ( Scope escopo, Operando <T> entrada, Operando <Integer> sourceTargetPairs)
Método de fábrica para criar uma classe envolvendo uma nova operação CollectivePermute.
Saída <T>
saída ()
A entrada permutada.

Métodos herdados

Métodos Públicos

public Output <T> asOutput ()

Retorna o identificador simbólico de um 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 <Integer> 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_pairs, 2].
Devoluções
  • uma nova instância de CollectivePermute

output public Output <T> output ()

A entrada permutada.