Struct RelationEmbedding 

pub struct RelationEmbedding {
    pub table: Array2<f64>,
    pub n_relations: usize,
    pub dim: usize,
}

Learnable embedding table for relation types.

Embeddings are initialised with Xavier uniform and can be updated via gradient-based methods (integration with an optimiser is the caller’s responsibility; this struct just holds the parameter tensor).

Fields

table: Array2<f64>

Embedding matrix (n_relations, dim).

n_relations: usize

Number of relation types.

dim: usize

Embedding dimensionality.

Implementations

impl RelationEmbedding

pub fn new(n_relations: usize, dim: usize) -> Self

Create a new relation embedding table with Xavier initialisation.

Arguments

• n_relations – Number of distinct relation types.
• dim – Embedding dimensionality.

pub fn get(&self, r: usize) -> Option<Array1<f64>>

Look up the embedding vector for relation r.

Returns None if r >= n_relations.

Trait Implementations

impl Clone for RelationEmbedding

fn clone(&self) -> RelationEmbedding

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for RelationEmbedding

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.