besskge.embedding.init_uniform_rotation

besskge.embedding.init_uniform_rotation(embedding_table)[source]

Initialize tensor with each entry being a uniformly distributed phase between 0 and \(2 \pi\). To be used for initialization of relation embedding tables in RotatE scoring function.

Parameters:

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

Return type:

Tensor

Returns:

Initialized tensor.