dgl.graphbolt.indptr_edge_ids

dgl.graphbolt.indptr_edge_ids(indptr, dtype=None, offset=None, output_size=None)[source]

将给定的 indptr 偏移量张量转换为边 ID 的 COO 格式张量。对于给定的 indptr [0, 2, 5, 7] 和 offset 张量 [0, 100, 200]，输出将是 [0, 1, 100, 101, 102, 201, 202]。如果未提供 offset，输出将是 [0, 1, 0, 1, 2, 0, 1]。

参数:

  • indptr (torch.Tensor) – 表示 csc_indptr 张量的一维张量。

  • dtype (Optional[torch.dtype]) – 返回的输出张量的数据类型。

  • offset (Optional[torch.Tensor]) – 表示将填充返回张量所用偏移量的一维张量。

  • output_size (Optional[int]) – 输出张量的大小。应等于 indptr[-1]。使用此参数可避免流同步以计算输出形状。

返回值:

转换后的 COO 边 ID 张量。

返回类型:

torch.Tensor