besskge.embedding.init_uniform_norm

besskge.embedding.init_uniform_norm(embedding_table)

Initialize embeddings according to uniform distribution and normalize so that each row has norm 1.

Parameters:

embedding_table (Tensor) – Tensor of embedding parameters to initialize.

Return type:

Tensor

Returns:

Initialized tensor.