KG2ESimilarity

class KG2ESimilarity(exact: bool = True)

基础类：Module, ABC

头部和尾部分布差异与关系分布之间的相似性。

仅针对具有对角协方差矩阵的多变量高斯分布实现。

初始化相似度模块。

Parameters:

exact (bool) – 是否返回精确的相似度，或者为了稍微提高速度而忽略常数偏移。

方法总结

forward(h, r, t)

计算相似度。

方法文档

abstract forward(h: GaussianDistribution, r: GaussianDistribution, t: GaussianDistribution) → Tensor

计算相似度。

# noqa: DAR401

Parameters:

• h (GaussianDistribution) – 形状: (*batch_dims, d) 头实体高斯分布。

• r (GaussianDistribution) – 形状: (*batch_dims, d) 关系高斯分布。

• t (GaussianDistribution) – 形状: (*batch_dims, d) 尾部实体高斯分布。

Returns:

torch.Tensor, 形状: (*batch_dims) # noqa: DAR202 相似度。

Return type:

Tensor