TorusE交互

class TorusEInteraction(p: int = 2, power_norm: bool = False)

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

来自[ebisu2018]的TorusE交互函数。

注意

这仅实现了基于两种L_p范数的变体。

初始化交互模块。

另请参阅

参数 p 和 power_norm 直接传递给 NormBasedInteraction。

Parameters:

• p (int) – 与 torch.linalg.vector_norm() 一起使用的范数。通常为1或2。

• power_norm (bool) – 是否使用\(L_p\)范数的p次方。它的优点是在0附近可微分，并且在数值上更稳定。

方法总结

forward(h, r, t)

评估交互函数。

方法文档

forward(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