CollapseGraphPairCombinator

class CollapseGraphPairCombinator[来源]

基础类：GraphPairCombinator

这个组合器将所有匹配的实体对合并为一个单一的ID。

方法总结

process(mapped_triples, alignment, offsets)

处理组合映射的三元组。

方法文档

process(mapped_triples: Tensor, alignment: Tensor, offsets: Tensor) → ProcessedTuple[source]

处理组合映射的三元组。

Parameters:

  • mapped_triples (Tensor) – 形状: (n, 3) 基于ID的合并三元组

  • alignment (Tensor) – 形状: (2, m) 基于ID的实体对齐

  • offsets (Tensor) – 形状: (2, 2) 合并后的实体和关系偏移量

Returns:

更新的三元组和对齐张量，以及用于更新标签到ID映射的参数

Return type:

ProcessedTuple