class DistMultInteraction[source]

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

A module wrapper for the stateless DistMult interaction function.

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

Methods Summary

func(r, t)

Evaluate the DistMult interaction function.

Methods Documentation

func(r, t)

Evaluate the DistMult interaction function.

  • h (FloatTensor) – shape: (batch_size, num_heads, 1, 1, dim) The head representations.

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

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

Return type



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