HolEInteraction

class HolEInteraction[source]

Bases: FunctionalInteraction[FloatTensor, FloatTensor, FloatTensor]

A module wrapper for the stateless HolE interaction function.

Initializes 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.