besskge.embedding.init_KGE_normal

besskge.embedding.init_KGE_normal(embedding_table, std=1.0, divide_by_embedding_size=True)

Initialize embeddings according to normal distribution with mean 0.

Parameters:

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

• std (float) – Standard deviation. Default: 1.0.

• divide_by_embedding_size (bool) – Rescale standard deviation by 1/row_size. Default: True.

Return type:

Tensor

Returns:

Initialized tensor.