DistMAInteraction

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

基础:FunctionalInteraction[Tensor, Tensor, Tensor]

来自[shi2019]的无状态DistMA交互函数。

对于头实体、关系和尾实体的表示 \(\mathbf{h}, \mathbf{r}, \mathbf{t} \in \mathbb{R}^d\), 交互函数由以下公式给出:

\[\langle \mathbf{h}, \mathbf{r}\rangle + \langle \mathbf{r}, \mathbf{t}\rangle + \langle \mathbf{h}, \mathbf{t}\rangle\]

注意

这个交互函数是对应论文中的对称部分 \(E_1\),而不是与 ComplExInteraction 的组合。

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

方法总结

func(h, r, t)

评估交互函数。

方法文档

static func(h: Tensor, r: Tensor, t: Tensor) Tensor[来源]

评估交互函数。

另请参阅

Interaction.forward 提供了关于交互函数通用批处理形式的详细描述。

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

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

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

Returns:

形状: batch_dims 分数。

Return type:

Tensor