QuatE交互

class QuatEInteraction

基础类: Interaction[Tensor, Tensor, Tensor]

无状态的QuatE交互函数。

它被给出为

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

其中 \(\mathbf{h}, \mathbf{r}, \mathbf{t} \in \mathbb{H}^d\) 是四元数表示， \(\otimes\) 表示哈密尔顿积，而 \(\langle \cdot, \cdot \rangle\) 表示内积。

警告

为了表示旋转，\(\mathbf{r}\) 必须被归一化为单位长度，参见 pykeen.nn.quaternion.normalize()。

另请参阅

• https://en.wikipedia.org/wiki/Quaternion

初始化交互模块。

属性摘要

entity_shape	实体表示的符号形状
relation_shape	关系表示的符号形状

方法总结

forward(h, r, t)

评估给定嵌入的QuatE交互函数。

属性文档

entity_shape: Sequence[str] = ('dk',)

实体表示的符号形状

relation_shape: Sequence[str] = ('dk',)

关系表示的符号形状

方法文档

forward(h: Tensor, r: tuple[Tensor, Tensor], t: Tensor) → Tensor

评估给定嵌入的QuatE交互函数。

嵌入必须具有可广播的形状。

另请参阅

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

Parameters:

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

• r (tuple[Tensor, Tensor]) – 形状: (*batch_dims, dim, 4) 头部表示。

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

Returns:

形状: (…) 分数。

Return type:

Tensor