[Feature] Add GraphDataLoader implementation (#2496)

* add graph dataloader

* add to doc

* fix

* fix

* fix docstring

* update according to torch default_collate

* add unittest

* fix

* fix lint

* fix
This commit is contained in:
Tong He
2021-01-07 13:38:21 +08:00
committed by GitHub
parent fa5f4d6aa7
commit e2e524df41
4 changed files with 144 additions and 2 deletions

View File

@@ -16,6 +16,7 @@ and an ``EdgeDataLoader`` for edge/link prediction task.
.. autoclass:: NodeDataLoader
.. autoclass:: EdgeDataLoader
.. autoclass:: GraphDataLoader
.. _api-dataloading-neighbor-sampling:
Neighbor Sampler