"projects/git@developer.sourcefind.cn:OpenDAS/pytorch3d.git" did not exist on "73dc109dba87d77ddf93395cf4d1578936b5db91"
Unverified Commit 4fd0a158 authored by caojy1998's avatar caojy1998 Committed by GitHub
Browse files

[CI] Fix a description bug for AddSelfLoop (#5842)


Co-authored-by: default avatarUbuntu <ubuntu@ip-172-31-22-182.ap-northeast-1.compute.internal>
parent a0512717
......@@ -544,7 +544,7 @@ class AddSelfLoop(BaseTransform):
>>> g.edata['he'] = torch.arange(3).float().reshape(-1, 1)
>>> new_g = transform(g)
>>> print(new_g.edges())
(tensor([1, 0, 1, 2]), tensor([2, 0, 1, 2]))
(tensor([0, 0, 2, 0, 1, 2]), tensor([2, 1, 0, 0, 1, 2]))
>>> print(new_g.edata('he'))
tensor([[0.],
[1.],
......@@ -569,7 +569,7 @@ class AddSelfLoop(BaseTransform):
>>> print(new_g.edges(etype='plays'))
(tensor([0, 1]), tensor([0, 1]))
>>> print(new_g.edges(etype='follows'))
(tensor([1, 2]), tensor([0, 1]))
(tensor([1, 2, 0, 1, 2]), tensor([0, 1, 0, 1, 2]))
>>> print(new_g.edata['feat'][('user', 'follows', 'user')].shape)
torch.Size([5, 5])
......@@ -578,7 +578,7 @@ class AddSelfLoop(BaseTransform):
>>> transform = AddSelfLoop(new_etypes=True)
>>> new_g = transform(g)
>>> print(new_g.edges(etype='follows'))
(tensor([1, 0, 1, 2]), tensor([2, 0, 1, 2]))
(tensor([1, 2, 0, 1, 2]), tensor([0, 1, 0, 1, 2]))
>>> print(new_g.edges(etype=('game', 'self', 'game')))
(tensor([0, 1]), tensor([0, 1]))
"""
......
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