".github/git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "a0542c1917b4530e5c49b4463159340fb1da66f4"
Unverified Commit 35e2d9a3 authored by yxy235's avatar yxy235 Committed by GitHub
Browse files

[GraphBolt] Modify docstring example of `InSubgraphSampler`. (#6822)


Co-authored-by: default avatarUbuntu <ubuntu@ip-172-31-0-133.us-west-2.compute.internal>
parent 208fa368
...@@ -44,13 +44,19 @@ class InSubgraphSampler(SubgraphSampler): ...@@ -44,13 +44,19 @@ class InSubgraphSampler(SubgraphSampler):
... print(data.sampled_subgraphs[0].node_pairs) ... print(data.sampled_subgraphs[0].node_pairs)
... print(data.sampled_subgraphs[0].original_row_node_ids) ... print(data.sampled_subgraphs[0].original_row_node_ids)
... print(data.sampled_subgraphs[0].original_column_node_ids) ... print(data.sampled_subgraphs[0].original_column_node_ids)
(tensor([0, 1, 2, 3, 4]), tensor([0, 0, 0, 1, 1])) CSCFormatBase(indptr=tensor([0, 3, 5]),
indices=tensor([0, 1, 2, 3, 4]),
)
tensor([0, 1, 4, 2, 3]) tensor([0, 1, 4, 2, 3])
tensor([0, 1]) tensor([0, 1])
(tensor([2, 3, 4, 0]), tensor([0, 0, 1, 1])) CSCFormatBase(indptr=tensor([0, 2, 4]),
indices=tensor([2, 3, 4, 0]),
)
tensor([2, 3, 0, 5, 1]) tensor([2, 3, 0, 5, 1])
tensor([2, 3]) tensor([2, 3])
(tensor([2, 3, 1, 4, 0]), tensor([0, 0, 0, 1, 1])) CSCFormatBase(indptr=tensor([0, 3, 5]),
indices=tensor([2, 3, 1, 4, 0]),
)
tensor([4, 5, 0, 3, 1]) tensor([4, 5, 0, 3, 1])
tensor([4, 5]) tensor([4, 5])
""" """
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment