HolEInteraction

class HolEInteraction(*args, **kwargs)[source]

Bases: FunctionalInteraction[FloatTensor, FloatTensor, FloatTensor]

A module wrapper for the stateless HolE interaction function.

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

Methods Summary

func(r, t)

Evaluate the HolE interaction function.

Methods Documentation

func(r, t)

Evaluate the HolE interaction function.

Parameters:
  • h (FloatTensor) – shape: (*batch_dims, dim) The head representations.

  • r (FloatTensor) – shape: (*batch_dims, dim) The relation representations.

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

Return type:

FloatTensor

Returns:

shape: batch_dims The scores.