Skip to content

Instantly share code, notes, and snippets.

@juliensimon

juliensimon/dgl2.py

Last active December 20, 2019 19:09
Show Gist options
  • Save juliensimon/26c48d9d56f891ea0f70cff4d4b51afb to your computer and use it in GitHub Desktop.
Save juliensimon/26c48d9d56f891ea0f70cff4d4b51afb to your computer and use it in GitHub Desktop.
Download ZIP
DGL part 2
Raw
dgl2.py
features = torch.eye(node_count)
print(features[2])
tensor([[0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,
0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.]])
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment