examples/pytorch/rgcn/layers.py · 52d4535b61cd82bc82a1ebb79d8dc66c5c50856e · OpenDAS / dgl · GitLab

Find file Blame History Permalink

[BugFix] Fix bug in RGCN data processing and use index_select to improve speed (#429) · d3c24cc2
Lingfan Yu authored Mar 04, 2019
```
* use index_select instead of __getitem__

* fix bug in dataset processing

* fix edge_type shape bug

* comments
```
d3c24cc2

layers.py 5.27 KB