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 próprio tensor é uma tupla aninhada XLA contendo N elementos (onde N é a razão do número de incorporação de núcleos de tensor por chip TPU). Cada elemento da tupla aninhada é uma tupla de tensores de nível 1. Cada tensor contém índices (DT_UINT32) para pesquisa de incorporação 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>
comoSaída ()
Retorna o identificador simbólico de um tensor.
static XlaRecvTPUEembeddingDeduplicationData
criar ( escopo do escopo, configuração de string)
Método de fábrica para criar uma classe que envolve uma nova operação XlaRecvTPUEmbeddingDeduplicationData.
Saída <?>
saída ()

Métodos Herdados

Métodos públicos

Public Output <Object> 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 de escopo, configuração de string)

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

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

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