QuatEInteraction

class QuatEInteraction

Bases: FunctionalInteraction[FloatTensor, FloatTensor, FloatTensor]

A module wrapper for the QuatE interaction function.

See also

pykeen.nn.functional.quat_e_interaction()

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

Methods Summary

func(r, t)

Evaluate the interaction function of QuatE for given embeddings.

Methods Documentation

func(r, t)

Evaluate the interaction function of QuatE for given embeddings.

The embeddings have to be in a broadcastable shape.

Note

dim has to be divisible by 4.

Parameters

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

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

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

Returns

shape: (…) The scores.