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)评估交互函数。
方法文档