class ComplExInteraction[source]

Bases: pykeen.nn.modules.FunctionalInteraction[torch.FloatTensor, torch.FloatTensor, torch.FloatTensor]

A module wrapper for the stateless ComplEx interaction function.

Initializes internal Module state, shared by both nn.Module and ScriptModule.

Methods Summary

func(r, t)

Evaluate the ComplEx interaction function.

Methods Documentation

func(r, t)

Evaluate the ComplEx interaction function.

\[Re(\langle h, r, conj(t) \rangle)\]
  • h (FloatTensor) – shape: (batch_size, num_heads, 1, 1, 2*dim) The complex head representations.

  • r (FloatTensor) – shape: (batch_size, 1, num_relations, 1, 2*dim) The complex relation representations.

  • t (FloatTensor) – shape: (batch_size, 1, 1, num_tails, 2*dim) The complex tail representations.

Return type



shape: (batch_size, num_heads, num_relations, num_tails) The scores.