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 | |
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