dmlc--dgl
Quan (Andy) Gan
ac53c1fae2
Co-authored-by: Rhett Ying <85214957+Rhett-Ying@users.noreply.github.com>
23 行
412 B
Python
23 行
412 B
Python
"""Subgraph samplers"""
|
|
|
|
from torchdata.datapipes.iter import Mapper
|
|
|
|
|
|
class SubgraphSampler(Mapper):
|
|
"""A subgraph sampler.
|
|
|
|
It is an iterator equivalent to the following:
|
|
|
|
.. code:: python
|
|
|
|
for data in datapipe:
|
|
yield sampler_func(data)
|
|
|
|
Parameters
|
|
----------
|
|
datapipe : DataPipe
|
|
The datapipe.
|
|
fn : callable
|
|
The subgraph sampling function.
|
|
"""
|