XlaRecvTPUEmbeddingDeduplicationData

classe final pública XlaRecvTPUEmbeddingDeduplicationData

Recebe dados de desduplicação (índices e pesos) do núcleo de incorporação.

Os dados de desduplicação são um Tensor com type=DT_VARIANT. O tensor em si é uma tupla aninhada XLA contendo N elementos (onde N é a razão entre o número de núcleos incorporados e tensores por chip TPU). Cada elemento da tupla aninhada é uma tupla de tensores de classificação 1. Cada tensor contém índices (DT_UINT32) para incorporar a pesquisa no TensorCore ou pesos (DT_FLOAT) para aplicar à saída da operação de pesquisa de incorporação.

Métodos Públicos

Saída <Objeto>
asOutput ()
Retorna o identificador simbólico de um tensor.
XlaRecvTPUEmbeddingDeduplicationData estático
create (escopo do escopo , configuração de string)
Método de fábrica para criar uma classe que agrupa uma nova operação XlaRecvTPUEmbeddingDeduplicationData.
Saída <?>
saída ()

Métodos herdados

Métodos Públicos

Saída pública <Objeto> 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 XlaRecvTPUEmbeddingDeduplicationData create (escopo do escopo , configuração de string)

Método de fábrica para criar uma classe que agrupa uma nova operação XlaRecvTPUEmbeddingDeduplicationData.

Parâmetros
escopo escopo atual
configuração Protocolo TPUEmbeddingConfiguration serializado.
Devoluções
  • uma nova instância de XlaRecvTPUEmbeddingDeduplicationData

Saída pública <?> saída ()