CollectivePermute

classe final pública CollectivePermute

Um Op para permutar tensores em instâncias de TPU replicadas.

Cada instância fornece sua própria entrada.

Por exemplo, suponha que haja 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]`.

Métodos públicos

Saída <T>	comoSaída () Retorna o identificador simbólico de um tensor.
estático <T> ColetivoPermute <T>	create ( Escopo do escopo, entrada Operando <T>, Operando <Integer> sourceTargetPairs) Método de fábrica para criar uma classe que envolve uma nova operação CollectivePermute.
Saída <T>	saída () A entrada permutada.

Métodos Herdados

Da classe org.tensorflow.op.PrimitiveOp

booleano final	igual a (objeto obj)
final int	código de hash ()
Operação	op () Retorna a Operation subjacente
String final	toString ()

Da classe java.lang.Object

boleano	igual a (objeto arg0)
Aula final<?>	getClass ()
int	código de hash ()
vazio final	notifique ()
vazio final	notifiqueTodos ()
Fragmento	toString ()
vazio final	espere (long arg0, int arg1)
vazio final	espere (arg0 longo)
vazio final	espere ()

Da interface org.tensorflow.Operand

Saída abstrata <T>

comoSaída () Retorna o identificador simbólico de um tensor.

Métodos públicos

Public Output <T> asOutput ()

Retorna o identificador simbólico de um 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 <Integer> sourceTargetPairs)

Método de fábrica para criar uma classe que envolve 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

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

A entrada permutada.