se_interaction

se_interaction(h, r_h, r_t, t, p, power_norm=False)[source]

Evaluate the Structured Embedding interaction function.

\[f(h, r, t) = -\|R_h h - R_t t\|\]
Parameters
  • h (FloatTensor) – shape: (*batch_dims, dim) The head representations.

  • r_h (FloatTensor) – shape: (*batch_dims, rel_dim, dim) The relation-specific head projection.

  • r_t (FloatTensor) – shape: (*batch_dims, rel_dim, dim) The relation-specific tail projection.

  • t (FloatTensor) – shape: (*batch_dims, dim) The tail representations.

  • p (int) – The p for the norm. cf. torch.linalg.vector_norm().

  • power_norm (bool) – Whether to return the powered norm.

Return type

FloatTensor

Returns

shape: batch_dims The scores.