1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
|
torch_geometric.transforms
==========================
.. contents:: Contents
:local:
Transforms are a general way to modify and customize :class:`~torch_geometric.data.Data` or :class:`~torch_geometric.data.HeteroData` objects, either by implicitly passing them as an argument to a :class:`~torch_geometric.data.Dataset`, or by applying them explicitly to individual :class:`~torch_geometric.data.Data` or :class:`~torch_geometric.data.HeteroData` objects:
.. code-block:: python
import torch_geometric.transforms as T
from torch_geometric.datasets import TUDataset
transform = T.Compose([T.ToUndirected(), T.AddSelfLoops()])
dataset = TUDataset(path, name='MUTAG', transform=transform)
data = dataset[0] # Implicitly transform data on every access.
data = TUDataset(path, name='MUTAG')[0]
data = transform(data) # Explicitly transform data.
General Transforms
------------------
.. currentmodule:: torch_geometric.transforms
.. autosummary::
:nosignatures:
:toctree: ../generated
:template: autosummary/only_class.rst
{% for name in torch_geometric.transforms.general_transforms %}
{{ name }}
{% endfor %}
Graph Transforms
----------------
.. currentmodule:: torch_geometric.transforms
.. autosummary::
:nosignatures:
:toctree: ../generated
:template: autosummary/only_class.rst
{% for name in torch_geometric.transforms.graph_transforms %}
{{ name }}
{% endfor %}
Vision Transforms
-----------------
.. currentmodule:: torch_geometric.transforms
.. autosummary::
:nosignatures:
:toctree: ../generated
:template: autosummary/only_class.rst
{% for name in torch_geometric.transforms.vision_transforms %}
{{ name }}
{% endfor %}
|
