RotatEInteraction

class RotatEInteraction[source]

Bases: FunctionalInteraction[FloatTensor, FloatTensor, FloatTensor]

A module wrapper for the stateless RotatE interaction function.

See also

pykeen.nn.functional.rotate_interaction()

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

Methods Summary

func(r, t)

Evaluate the RotatE interaction function.

Methods Documentation

func(r, t)

Evaluate the RotatE interaction function.

Note

this method expects all tensors to be of complex datatype, i.e., torch.is_complex(x) to evaluate to True.

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.