RotatEInteraction

class RotatEInteraction(*args, **kwargs)

基础：FunctionalInteraction[Tensor, Tensor, Tensor]

由[sun2019]提出的RotatE交互函数。

RotatE 在复数值的实体和关系表示上操作，即， \(\textbf{e}_i, \textbf{r}_i \in \mathbb{C}^d\)。

注意

此方法通常期望所有张量都是复数数据类型，即torch.is_complex(x)评估为True。然而，为了向后兼容和使用方便，您也可以传递形状符合torch.view_as_complex()的实数张量，参见pykeen.utils.ensure_complex()。

初始化内部模块状态，由nn.Module和ScriptModule共享。

属性摘要

is_complex

交互是否在复杂输入上定义

方法总结

func(h, r, t)

评估交互函数。

属性文档

is_complex: ClassVar[bool] = True

交互是否在复杂输入上定义

方法文档

static func(h: Tensor, r: Tensor, t: Tensor) → Tensor

评估交互函数。

注意

此方法期望所有张量都是复数数据类型，即torch.is_complex(x)评估为True。

Parameters:

• h (Tensor) – 形状: (*batch_dims, dim) 头部表示。

• r (Tensor) – 形状: (*batch_dims, dim) 关系表示。

• t (Tensor) – 形状: (*batch_dims, dim) 尾部表示。

Returns:

形状: batch_dims 分数。

Return type:

Tensor