2.0.2
Notes
Installation
Quick Start
Installation via Anaconda
Installation via Pip Wheels
Installation from Source
Frequently Asked Questions
Introduction by Example
Data Handling of Graphs
Common Benchmark Datasets
Mini-batches
Data Transforms
Learning Methods on Graphs
Creating Message Passing Networks
The “MessagePassing” Base Class
Implementing the GCN Layer
Implementing the Edge Convolution
Creating Your Own Datasets
Creating “In Memory Datasets”
Creating “Larger” Datasets
Frequently Asked Questions
Heterogeneous Graph Learning
Example Graph
Creating Heterogeneous Graphs
Heterogeneous Graph Transformations
Creating Heterogeneous GNNs
Heterogeneous Graph Samplers
Loading Graphs from CSV
Managing Experiments with GraphGym
Highlights
Why GraphGym?
Basic Usage
In-Depth Usage
Customizing GraphGym
Advanced Mini-Batching
Pairs of Graphs
Bipartite Graphs
Batching Along New Dimensions
Memory-Efficient Aggregations
TorchScript Support
Converting GNN Models
Creating Jittable GNN Operators
GNN Cheatsheet
Graph Neural Network Operators
Heterogeneous Graph Neural Network Operators
Hypergraph Neural Network Operators
Point Cloud Neural Network Operators
Colab Notebooks and Video Tutorials
External Resources
Package Reference
torch_geometric
torch_geometric.nn
Convolutional Layers
Dense Convolutional Layers
Normalization Layers
Global Pooling Layers
Pooling Layers
Dense Pooling Layers
Unpooling Layers
Models
Functional
Model Transformations
DataParallel Layers
torch_geometric.data
torch_geometric.loader
torch_geometric.datasets
torch_geometric.transforms
torch_geometric.utils
torch_geometric.graphgym
Workflow Modules
Model Modules
Register Modules
Utility Modules
torch_geometric.profile
pytorch_geometric
»
Index
Edit on GitHub
Index
_
|
A
|
B
|
C
|
D
|
E
|
F
|
G
|
H
|
I
|
J
|
K
|
L
|
M
|
N
|
O
|
P
|
Q
|
R
|
S
|
T
|
U
|
V
|
W
|
X
|
Y
|
Z
_
__cat_dim__() (Data method)
(HeteroData method)
__inc__() (Data method)
(HeteroData method)
A
accuracy() (in module torch_geometric.utils)
Actor (class in torch_geometric.datasets)
add_remaining_self_loops() (in module torch_geometric.utils)
add_self_loops() (in module torch_geometric.utils)
AddMetaPaths (class in torch_geometric.transforms)
AddSelfLoops (class in torch_geometric.transforms)
affine (InstanceNorm attribute)
agg_batch() (in module torch_geometric.graphgym.utils)
agg_runs() (in module torch_geometric.graphgym.utils)
aggregate() (MessagePassing method)
AGNNConv (class in torch_geometric.nn.conv)
Airports (class in torch_geometric.datasets)
Amazon (class in torch_geometric.datasets)
AmazonProducts (class in torch_geometric.datasets)
AMiner (class in torch_geometric.datasets)
apply() (Data method)
(HeteroData method)
apply_() (Data method)
(HeteroData method)
APPNP (class in torch_geometric.nn.conv)
ARGA (class in torch_geometric.nn.models)
ARGVA (class in torch_geometric.nn.models)
ARMAConv (class in torch_geometric.nn.conv)
ASAPooling (class in torch_geometric.nn.pool)
AtomEncoder (class in torch_geometric.graphgym.models)
att_x (SuperGATConv attribute)
att_y (SuperGATConv attribute)
AttentiveFP (class in torch_geometric.nn.models)
AttributedGraphDataset (class in torch_geometric.datasets)
augment_adj() (GraphUNet method)
auto_select_device() (in module torch_geometric.graphgym.utils)
avg_pool() (in module torch_geometric.nn.pool)
avg_pool_neighbor_x() (in module torch_geometric.nn.pool)
avg_pool_x() (in module torch_geometric.nn.pool)
B
barabasi_albert_graph() (in module torch_geometric.utils)
BaseTransform (class in torch_geometric.transforms)
Batch (class in torch_geometric.data)
batched_negative_sampling() (in module torch_geometric.utils)
BatchNorm (class in torch_geometric.nn.norm)
BatchNorm1dEdge (class in torch_geometric.graphgym.models)
BatchNorm1dNode (class in torch_geometric.graphgym.models)
BitcoinOTC (class in torch_geometric.datasets)
BondEncoder (class in torch_geometric.graphgym.models)
bro() (in module torch_geometric.nn.functional)
C
call_function() (Transformer method)
call_message_passing_module() (Transformer method)
call_method() (Transformer method)
call_module() (Transformer method)
Cartesian (class in torch_geometric.transforms)
Center (class in torch_geometric.transforms)
CGConv (class in torch_geometric.nn.conv)
ChebConv (class in torch_geometric.nn.conv)
CitationFull (class in torch_geometric.datasets)
clean_ckpt() (in module torch_geometric.graphgym)
clone() (Data method)
(HeteroData method)
ClusterData (class in torch_geometric.loader)
ClusterGCNConv (class in torch_geometric.nn.conv)
ClusterLoader (class in torch_geometric.loader)
coalesce() (Data method)
(HeteroData method)
(in module torch_geometric.utils)
Coauthor (class in torch_geometric.datasets)
coeffs (GNNExplainer attribute)
collate() (InMemoryDataset static method)
collect() (HeteroData method)
CoMA (class in torch_geometric.datasets)
Compose (class in torch_geometric.transforms)
compute_edge_score_sigmoid() (EdgePooling static method)
compute_edge_score_softmax() (EdgePooling static method)
compute_edge_score_tanh() (EdgePooling static method)
compute_loss() (in module torch_geometric.graphgym)
Constant (class in torch_geometric.transforms)
contains_isolated_nodes() (in module torch_geometric.utils)
contains_self_loops() (in module torch_geometric.utils)
contiguous() (Data method)
(HeteroData method)
copy() (InMemoryDataset method)
CoraFull (class in torch_geometric.datasets)
correct() (CorrectAndSmooth method)
CorrectAndSmooth (class in torch_geometric.nn.models)
count_parameters() (in module torch_geometric.profile)
cpu() (Data method)
(HeteroData method)
create_loader() (in module torch_geometric.graphgym)
create_logger() (in module torch_geometric.graphgym)
create_model() (in module torch_geometric.graphgym)
create_optimizer() (in module torch_geometric.graphgym)
create_scheduler() (in module torch_geometric.graphgym)
create_spectral_features() (SignedGCN method)
cuda() (Data method)
(HeteroData method)
D
Data (class in torch_geometric.data)
DataListLoader (class in torch_geometric.loader)
DataLoader (class in torch_geometric.loader)
DataParallel (class in torch_geometric.nn.data_parallel)
Dataset (class in torch_geometric.data)
DBLP (class in torch_geometric.datasets)
DBP15K (class in torch_geometric.datasets)
debug (class in torch_geometric.debug)
decode() (GAE method)
DeepGCNLayer (class in torch_geometric.nn.models)
DeepGraphInfomax (class in torch_geometric.nn.models)
DeezerEurope (class in torch_geometric.datasets)
degree() (in module torch_geometric.utils)
Delaunay (class in torch_geometric.transforms)
dense_diff_pool (class in torch_geometric.nn.dense)
dense_mincut_pool (class in torch_geometric.nn.dense)
dense_to_sparse() (in module torch_geometric.utils)
DenseDataLoader (class in torch_geometric.loader)
DenseGCNConv (class in torch_geometric.nn.dense)
DenseGINConv (class in torch_geometric.nn.dense)
DenseGraphConv (class in torch_geometric.nn.dense)
DenseSAGEConv (class in torch_geometric.nn.dense)
detach() (Data method)
(HeteroData method)
(TGNMemory method)
detach_() (Data method)
(HeteroData method)
dict_list_to_json() (in module torch_geometric.graphgym.utils)
dict_to_json() (in module torch_geometric.graphgym.utils)
dict_to_tb() (in module torch_geometric.graphgym.utils)
DiffGroupNorm (class in torch_geometric.nn.norm)
diffusion_matrix_approx() (GDC method)
diffusion_matrix_exact() (GDC method)
DimeNet (class in torch_geometric.nn.models)
discriminate() (DeepGraphInfomax method)
(SignedGCN method)
discriminator_loss() (ARGA method)
Distance (class in torch_geometric.transforms)
DNAConv (class in torch_geometric.nn.conv)
download() (Dataset method)
(InMemoryDataset method)
download_url() (in module torch_geometric.data)
dropout_adj() (in module torch_geometric.utils)
dummy_context (class in torch_geometric.graphgym.utils)
dump_cfg() (in module torch_geometric.graphgym)
DynamicEdgeConv (class in torch_geometric.nn.conv)
DynamicFAUST (class in torch_geometric.datasets)
E
ECConv (in module torch_geometric.nn.conv)
edge_stores (HeteroData property)
edge_types (HeteroData property)
EdgeConv (class in torch_geometric.nn.conv)
EdgePooling (class in torch_geometric.nn.pool)
EGConv (class in torch_geometric.nn.conv)
embed() (RECT_L method)
encode() (ARGVA method)
(GAE method)
(VGAE method)
Entities (class in torch_geometric.datasets)
eps (InstanceNorm attribute)
erdos_renyi_graph() (in module torch_geometric.utils)
explain_graph() (GNNExplainer method)
explain_node() (GNNExplainer method)
extract_bz2() (in module torch_geometric.data)
extract_gz() (in module torch_geometric.data)
extract_tar() (in module torch_geometric.data)
extract_zip() (in module torch_geometric.data)
F
f1_score() (in module torch_geometric.utils)
FacebookPagePage (class in torch_geometric.datasets)
FaceToEdge (class in torch_geometric.transforms)
FAConv (class in torch_geometric.nn.conv)
false_negative() (in module torch_geometric.utils)
false_positive() (in module torch_geometric.utils)
FastRGCNConv (class in torch_geometric.nn.conv)
FAUST (class in torch_geometric.datasets)
FeaStConv (class in torch_geometric.nn.conv)
FeatureEncoder (class in torch_geometric.graphgym.models)
FiLMConv (class in torch_geometric.nn.conv)
FixedPoints (class in torch_geometric.transforms)
Flickr (class in torch_geometric.datasets)
forward() (AGNNConv method)
(APPNP method)
(ARMAConv method)
(ASAPooling method)
(AttentiveFP method)
(BatchNorm method)
(CGConv method)
(ChebConv method)
(ClusterGCNConv method)
(DataParallel method)
(DeepGCNLayer method)
(DeepGraphInfomax method)
(DenseGCNConv method)
(DenseGINConv method)
(DenseGraphConv method)
(DenseSAGEConv method)
(DiffGroupNorm method)
(DimeNet method)
(DNAConv method)
(DynamicEdgeConv method)
(EdgeConv method)
(EdgePooling method)
(EGConv method)
(FAConv method)
(FastRGCNConv method)
(FeaStConv method)
(FiLMConv method)
(GATConv method)
(GatedGraphConv method)
(GATv2Conv method)
(GCN2Conv method)
(GCNConv method)
(GENConv method)
(GeneralConv method)
(GINConv method)
(GINEConv method)
(GlobalAttention method)
(GMMConv method)
(GraphConv method)
(GraphMultisetTransformer method)
(GraphNorm method)
(GraphSizeNorm method)
(GraphUNet method)
(GravNetConv method)
(HEATConv method)
(HeteroConv method)
(HeteroLinear method)
(HGTConv method)
(HypergraphConv method)
(InnerProductDecoder method)
(InstanceNorm method)
(JumpingKnowledge method)
(LabelPropagation method)
(LayerNorm method)
(LEConv method)
(Linear method)
(MemPooling method)
(MessageNorm method)
(MetaLayer method)
(MetaPath2Vec method)
(MFConv method)
(NNConv method)
(Node2Vec method)
(PairNorm method)
(PANConv method)
(PANPooling method)
(PDNConv method)
(PNAConv method)
(PointNetConv method)
(PointTransformerConv method)
(PPFConv method)
(RECT_L method)
(RENet method)
(ResGatedGraphConv method)
(RGCNConv method)
(SAGEConv method)
(SAGPooling method)
(SchNet method)
(Set2Set method)
(SGConv method)
(SignedConv method)
(SignedGCN method)
(SplineConv method)
(SuperGATConv method)
(TAGConv method)
(TGNMemory method)
(TopKPooling method)
(TransformerConv method)
(WLConv method)
(XConv method)
forward_all() (InnerProductDecoder method)
fps() (in module torch_geometric.nn.pool)
from_data_list() (Batch class method)
from_dict() (Data class method)
from_networkx() (in module torch_geometric.utils)
from_qm9_pretrained() (DimeNet static method)
(SchNet static method)
from_scipy_sparse_matrix() (in module torch_geometric.utils)
from_trimesh() (in module torch_geometric.utils)
G
GAE (class in torch_geometric.nn.models)
GAT (class in torch_geometric.nn.models)
GATConv (class in torch_geometric.graphgym.models)
(class in torch_geometric.nn.conv)
GatedGraphConv (class in torch_geometric.nn.conv)
GATv2Conv (class in torch_geometric.nn.conv)
GCN (class in torch_geometric.nn.models)
GCN2Conv (class in torch_geometric.nn.conv)
GCNConv (class in torch_geometric.graphgym.models)
(class in torch_geometric.nn.conv)
GCNNorm (class in torch_geometric.transforms)
GDC (class in torch_geometric.transforms)
GDELT (class in torch_geometric.datasets)
GEDDataset (class in torch_geometric.datasets)
GemsecDeezer (class in torch_geometric.datasets)
GENConv (class in torch_geometric.nn.conv)
GeneralConv (class in torch_geometric.graphgym.models)
(class in torch_geometric.nn.conv)
GeneralEdgeConv (class in torch_geometric.graphgym.models)
GeneralLayer (class in torch_geometric.graphgym.models)
GeneralMultiLayer (class in torch_geometric.graphgym.models)
GeneralSampleEdgeConv (class in torch_geometric.graphgym.models)
GenerateMeshNormals (class in torch_geometric.transforms)
geodesic_distance() (in module torch_geometric.utils)
GeometricShapes (class in torch_geometric.datasets)
get() (Dataset method)
(InMemoryDataset method)
get_attention() (SuperGATConv method)
get_attention_loss() (SuperGATConv method)
get_attr() (Transformer method)
get_cpu_memory_from_gc() (in module torch_geometric.profile)
get_current_gpu_usage() (in module torch_geometric.graphgym.utils)
get_data_size() (in module torch_geometric.profile)
get_edge_store() (HeteroData method)
get_example() (Batch method)
get_gpu_memory_from_gc() (in module torch_geometric.profile)
get_gpu_memory_from_nvidia_smi() (in module torch_geometric.profile)
get_laplacian() (in module torch_geometric.utils)
get_model_size() (in module torch_geometric.profile)
get_node_store() (HeteroData method)
get_semantic_labels() (RECT_L method)
get_stats_summary() (in module torch_geometric.profile)
GIN (class in torch_geometric.nn.models)
GINConv (class in torch_geometric.graphgym.models)
(class in torch_geometric.nn.conv)
GINEConv (class in torch_geometric.nn.conv)
gini() (in module torch_geometric.nn.functional)
GitHub (class in torch_geometric.datasets)
global_add_pool() (in module torch_geometric.graphgym.models)
(in module torch_geometric.nn.glob)
global_max_pool() (in module torch_geometric.graphgym.models)
(in module torch_geometric.nn.glob)
global_mean_pool() (in module torch_geometric.graphgym.models)
(in module torch_geometric.nn.glob)
global_sort_pool() (in module torch_geometric.nn.glob)
GlobalAttention (class in torch_geometric.nn.glob)
GMMConv (class in torch_geometric.nn.conv)
GNN (class in torch_geometric.graphgym.models)
GNNBenchmarkDataset (class in torch_geometric.datasets)
GNNEdgeHead (class in torch_geometric.graphgym.models)
GNNExplainer (class in torch_geometric.nn.models)
GNNGraphHead (class in torch_geometric.graphgym.models)
GNNLayer() (in module torch_geometric.graphgym.models)
GNNNodeHead (class in torch_geometric.graphgym.models)
GNNPreMP() (in module torch_geometric.graphgym.models)
GNNStackStage (class in torch_geometric.graphgym.models)
graclus() (in module torch_geometric.nn.pool)
GraphConv (class in torch_geometric.nn.conv)
GraphMultisetTransformer (class in torch_geometric.nn.glob)
GraphNorm (class in torch_geometric.nn.norm)
GraphSAGE (class in torch_geometric.nn.models)
GraphSAINTEdgeSampler (class in torch_geometric.loader)
GraphSAINTNodeSampler (class in torch_geometric.loader)
GraphSAINTRandomWalkSampler (class in torch_geometric.loader)
GraphSAINTSampler (class in torch_geometric.loader)
GraphSizeNorm (class in torch_geometric.nn.norm)
GraphUNet (class in torch_geometric.nn.models)
GravNetConv (class in torch_geometric.nn.conv)
grid() (in module torch_geometric.utils)
GridSampling (class in torch_geometric.transforms)
group_distance_ratio() (DiffGroupNorm static method)
H
has_edge_level_arg() (Transformer method)
has_isolated_nodes() (Data method)
(HeteroData method)
has_self_loops() (Data method)
(HeteroData method)
HEATConv (class in torch_geometric.nn.conv)
HeteroConv (class in torch_geometric.nn.conv)
HeteroData (class in torch_geometric.data)
HeteroLinear (class in torch_geometric.nn.dense)
HGTConv (class in torch_geometric.nn.conv)
HGTLoader (class in torch_geometric.loader)
histogram() (WLConv method)
homophily() (in module torch_geometric.utils)
HypergraphConv (class in torch_geometric.nn.conv)
I
ICEWS18 (class in torch_geometric.datasets)
IMDB (class in torch_geometric.datasets)
index_select() (Batch method)
(Dataset method)
init_submodule() (Transformer method)
init_weights() (in module torch_geometric.graphgym)
InMemoryDataset (class in torch_geometric.data)
InnerProductDecoder (class in torch_geometric.nn.models)
InstanceNorm (class in torch_geometric.nn.norm)
IntegerFeatureEncoder (class in torch_geometric.graphgym.models)
intersection_and_union() (in module torch_geometric.utils)
is_ckpt_epoch() (in module torch_geometric.graphgym.utils)
is_coalesced() (Data method)
(HeteroData method)
is_debug_enabled() (in module torch_geometric.debug)
is_directed() (Data method)
(HeteroData method)
is_edge_attr() (Data method)
is_edge_level() (Transformer method)
is_eval_epoch() (in module torch_geometric.graphgym.utils)
is_node_attr() (Data method)
is_undirected() (Data method)
(HeteroData method)
(in module torch_geometric.utils)
J
jittable() (MessagePassing method)
JODIEDataset (class in torch_geometric.datasets)
JumpingKnowledge (class in torch_geometric.nn.models)
K
k_hop_subgraph() (in module torch_geometric.utils)
KarateClub (class in torch_geometric.datasets)
keys (Data property)
(HeteroData property)
kl_loss() (ARGVA method)
(MemPooling static method)
(VGAE method)
knn() (in module torch_geometric.nn.pool)
knn_graph() (in module torch_geometric.nn.pool)
knn_interpolate() (in module torch_geometric.nn.unpool)
KNNGraph (class in torch_geometric.transforms)
L
LabelPropagation (class in torch_geometric.nn.models)
LaplacianLambdaMax (class in torch_geometric.transforms)
LastFM (class in torch_geometric.datasets)
LastFMAsia (class in torch_geometric.datasets)
LayerNorm (class in torch_geometric.nn.norm)
LEConv (class in torch_geometric.nn.conv)
len() (Dataset method)
(InMemoryDataset method)
(TrackMLParticleTrackingDataset method)
Linear (class in torch_geometric.graphgym.models)
(class in torch_geometric.nn.dense)
LinearTransformation (class in torch_geometric.transforms)
LineGraph (class in torch_geometric.transforms)
load_cfg() (in module torch_geometric.graphgym)
load_ckpt() (in module torch_geometric.graphgym)
loader() (MetaPath2Vec method)
(Node2Vec method)
LocalCartesian (class in torch_geometric.transforms)
LocalDegreeProfile (class in torch_geometric.transforms)
loss() (DeepGraphInfomax method)
(MetaPath2Vec method)
(Node2Vec method)
(SignedGCN method)
M
makedirs_rm_exist() (in module torch_geometric.graphgym.utils)
match_baseline_cfg() (in module torch_geometric.graphgym.utils)
max_pool() (in module torch_geometric.nn.pool)
max_pool_neighbor_x() (in module torch_geometric.nn.pool)
max_pool_x() (in module torch_geometric.nn.pool)
MD17 (class in torch_geometric.datasets)
mean_iou() (in module torch_geometric.utils)
MemPooling (class in torch_geometric.nn.pool)
message() (MessagePassing method)
message_and_aggregate() (MessagePassing method)
message_basic() (GeneralConv method)
MessageNorm (class in torch_geometric.nn.norm)
MessagePassing (class in torch_geometric.nn.conv.message_passing)
metadata() (HeteroData method)
MetaLayer (class in torch_geometric.nn.meta)
MetaPath2Vec (class in torch_geometric.nn.models)
MFConv (class in torch_geometric.nn.conv)
MixHopSyntheticDataset (class in torch_geometric.datasets)
MLP (class in torch_geometric.graphgym.models)
MLP() (GIN static method)
MNISTSuperpixels (class in torch_geometric.datasets)
ModelNet (class in torch_geometric.datasets)
module
torch_geometric.data
torch_geometric.datasets
torch_geometric.debug
torch_geometric.graphgym
torch_geometric.graphgym.models
torch_geometric.graphgym.utils
torch_geometric.loader
torch_geometric.nn.conv
torch_geometric.nn.conv.message_passing
torch_geometric.nn.data_parallel
torch_geometric.nn.functional
torch_geometric.nn.glob
torch_geometric.nn.meta
torch_geometric.nn.models
torch_geometric.nn.norm
torch_geometric.nn.pool
torch_geometric.nn.unpool
torch_geometric.profile
torch_geometric.seed
torch_geometric.transforms
torch_geometric.utils
MoleculeNet (class in torch_geometric.datasets)
momentum (InstanceNorm attribute)
N
nearest() (in module torch_geometric.nn.pool)
neg_embedding_loss() (SignedGCN method)
neg_sample() (Node2Vec method)
negative_sampling() (in module torch_geometric.utils)
(SuperGATConv method)
NeighborLoader (class in torch_geometric.loader)
NeighborSampler (class in torch_geometric.loader)
NELL (class in torch_geometric.datasets)
nll_loss() (SignedGCN method)
NNConv (class in torch_geometric.nn.conv)
Node2Vec (class in torch_geometric.nn.models)
node_stores (HeteroData property)
node_types (HeteroData property)
normalized_cut() (in module torch_geometric.utils)
NormalizeFeatures (class in torch_geometric.transforms)
NormalizeRotation (class in torch_geometric.transforms)
NormalizeScale (class in torch_geometric.transforms)
num_classes (InMemoryDataset property)
num_edge_features (Data property)
(Dataset property)
num_edges (Data property)
(HeteroData property)
num_faces (Data property)
num_features (Data property)
(Dataset property)
(InstanceNorm attribute)
num_graphs (Batch property)
num_hops (GNNExplainer property)
num_node_features (Data property)
(Dataset property)
num_nodes (Data property)
(HeteroData property)
O
OGB_MAG (class in torch_geometric.datasets)
OneHotDegree (class in torch_geometric.transforms)
output() (Transformer method)
P
PairNorm (class in torch_geometric.nn.norm)
PANConv (class in torch_geometric.nn.conv)
panentropy() (PANConv method)
PANPooling (class in torch_geometric.nn.pool)
params_count() (in module torch_geometric.graphgym.utils)
parse_args() (in module torch_geometric.graphgym)
PascalPF (class in torch_geometric.datasets)
PascalVOCKeypoints (class in torch_geometric.datasets)
PCPNetDataset (class in torch_geometric.datasets)
PDNConv (class in torch_geometric.nn.conv)
pin_memory() (Data method)
(HeteroData method)
placeholder() (Transformer method)
Planetoid (class in torch_geometric.datasets)
PNA (class in torch_geometric.nn.models)
PNAConv (class in torch_geometric.nn.conv)
PointConv (in module torch_geometric.nn.conv)
PointNetConv (class in torch_geometric.nn.conv)
PointPairFeatures (class in torch_geometric.transforms)
PointTransformerConv (class in torch_geometric.nn.conv)
Polar (class in torch_geometric.transforms)
pos_embedding_loss() (SignedGCN method)
pos_sample() (Node2Vec method)
positive_sampling() (SuperGATConv method)
PPFConv (class in torch_geometric.nn.conv)
PPI (class in torch_geometric.datasets)
pre_transform() (RENet static method)
precision() (in module torch_geometric.utils)
process() (Dataset method)
(InMemoryDataset method)
processed_file_names (Dataset property)
(InMemoryDataset property)
processed_paths (Dataset property)
profileit() (in module torch_geometric.profile)
propagate() (MessagePassing method)
Q
QM7b (class in torch_geometric.datasets)
QM9 (class in torch_geometric.datasets)
R
radius() (in module torch_geometric.nn.pool)
radius_graph() (in module torch_geometric.nn.pool)
RadiusGraph (class in torch_geometric.transforms)
RandomFlip (class in torch_geometric.transforms)
RandomLinkSplit (class in torch_geometric.transforms)
RandomNodeSampler (class in torch_geometric.loader)
RandomNodeSplit (class in torch_geometric.transforms)
RandomRotate (class in torch_geometric.transforms)
RandomScale (class in torch_geometric.transforms)
RandomShear (class in torch_geometric.transforms)
RandomTranslate (class in torch_geometric.transforms)
raw_file_names (Dataset property)
(InMemoryDataset property)
raw_paths (Dataset property)
recall() (in module torch_geometric.utils)
recon_loss() (GAE method)
record_stream() (Data method)
(HeteroData method)
RECT_L (class in torch_geometric.nn.models)
Reddit (class in torch_geometric.datasets)
Reddit2 (class in torch_geometric.datasets)
reg_loss() (ARGA method)
register_act() (in module torch_geometric.graphgym)
register_aggregate_forward_hook() (MessagePassing method)
register_aggregate_forward_pre_hook() (MessagePassing method)
register_base() (in module torch_geometric.graphgym)
register_config() (in module torch_geometric.graphgym)
register_edge_encoder() (in module torch_geometric.graphgym)
register_head() (in module torch_geometric.graphgym)
register_layer() (in module torch_geometric.graphgym)
register_loader() (in module torch_geometric.graphgym)
register_loss() (in module torch_geometric.graphgym)
register_message_and_aggregate_forward_hook() (MessagePassing method)
register_message_and_aggregate_forward_pre_hook() (MessagePassing method)
register_message_forward_hook() (MessagePassing method)
register_message_forward_pre_hook() (MessagePassing method)
register_network() (in module torch_geometric.graphgym)
register_node_encoder() (in module torch_geometric.graphgym)
register_optimizer() (in module torch_geometric.graphgym)
register_pooling() (in module torch_geometric.graphgym)
register_propagate_forward_hook() (MessagePassing method)
register_propagate_forward_pre_hook() (MessagePassing method)
register_scheduler() (in module torch_geometric.graphgym)
register_stage() (in module torch_geometric.graphgym)
register_train() (in module torch_geometric.graphgym)
RelLinkPredDataset (class in torch_geometric.datasets)
remove_isolated_nodes() (in module torch_geometric.utils)
remove_self_loops() (in module torch_geometric.utils)
RemoveIsolatedNodes (class in torch_geometric.transforms)
RemoveTrainingClasses (class in torch_geometric.transforms)
RENet (class in torch_geometric.nn.models)
reparametrize() (ARGVA method)
(VGAE method)
replace_all_uses_with() (Transformer method)
requires_grad_() (Data method)
(HeteroData method)
reset_parameters() (AGNNConv method)
(APPNP method)
(ARGA method)
(ARMAConv method)
(ASAPooling method)
(AttentiveFP method)
(BatchNorm method)
(CGConv method)
(ChebConv method)
(ClusterGCNConv method)
(DeepGCNLayer method)
(DeepGraphInfomax method)
(DenseGCNConv method)
(DenseGINConv method)
(DenseGraphConv method)
(DenseSAGEConv method)
(DiffGroupNorm method)
(DimeNet method)
(DNAConv method)
(DynamicEdgeConv method)
(EdgeConv method)
(EdgePooling method)
(EGConv method)
(FAConv method)
(FeaStConv method)
(FiLMConv method)
(GAE method)
(GATConv method)
(GatedGraphConv method)
(GATv2Conv method)
(GCN2Conv method)
(GCNConv method)
(GENConv method)
(GeneralConv method)
(GINConv method)
(GINEConv method)
(GlobalAttention method)
(GMMConv method)
(GraphConv method)
(GraphMultisetTransformer method)
(GraphNorm method)
(GraphUNet method)
(GravNetConv method)
(HEATConv method)
(HeteroConv method)
(HeteroLinear method)
(HGTConv method)
(HypergraphConv method)
(JumpingKnowledge method)
(LayerNorm method)
(LEConv method)
(Linear method)
(MemPooling method)
(MessageNorm method)
(MetaLayer method)
(MetaPath2Vec method)
(MFConv method)
(NNConv method)
(Node2Vec method)
(PANConv method)
(PANPooling method)
(PDNConv method)
(PNAConv method)
(PointNetConv method)
(PointTransformerConv method)
(PPFConv method)
(RECT_L method)
(RENet method)
(ResGatedGraphConv method)
(RGCNConv method)
(SAGEConv method)
(SAGPooling method)
(SchNet method)
(Set2Set method)
(SGConv method)
(SignedConv method)
(SignedGCN method)
(SplineConv method)
(SuperGATConv method)
(TAGConv method)
(TGNMemory method)
(TopKPooling method)
(TransformerConv method)
(WLConv method)
(XConv method)
reset_state() (TGNMemory method)
ResGatedGraphConv (class in torch_geometric.nn.conv)
RGCNConv (class in torch_geometric.nn.conv)
S
S3DIS (class in torch_geometric.datasets)
SAGEConv (class in torch_geometric.graphgym.models)
(class in torch_geometric.nn.conv)
SAGPooling (class in torch_geometric.nn.pool)
sample() (Node2Vec method)
SamplePoints (class in torch_geometric.transforms)
save_ckpt() (in module torch_geometric.graphgym)
scatter() (DataParallel method)
SchNet (class in torch_geometric.nn.models)
seed_everything() (in module torch_geometric.seed)
segregate_self_loops() (in module torch_geometric.utils)
Sequential (class in torch_geometric.nn.sequential)
Set2Set (class in torch_geometric.nn.glob)
set_agg_dir() (in module torch_geometric.graphgym)
set_cfg() (in module torch_geometric.graphgym)
set_debug (class in torch_geometric.debug)
set_printing() (in module torch_geometric.graphgym)
set_run_dir() (in module torch_geometric.graphgym)
SGConv (class in torch_geometric.nn.conv)
ShaDowKHopSampler (class in torch_geometric.loader)
ShapeNet (class in torch_geometric.datasets)
share_memory_() (Data method)
(HeteroData method)
SHREC2016 (class in torch_geometric.datasets)
shuffle() (Dataset method)
SIGN (class in torch_geometric.transforms)
SignedConv (class in torch_geometric.nn.conv)
SignedGCN (class in torch_geometric.nn.models)
size() (Data method)
(HeteroData method)
smooth() (CorrectAndSmooth method)
SNAPDataset (class in torch_geometric.datasets)
sort_edge_index() (in module torch_geometric.utils)
sparsify_dense() (GDC method)
sparsify_sparse() (GDC method)
Spherical (class in torch_geometric.transforms)
SplineConv (class in torch_geometric.graphgym.models)
(class in torch_geometric.nn.conv)
split_edges() (SignedGCN method)
stochastic_blockmodel_graph() (in module torch_geometric.utils)
stores (HeteroData property)
structured_negative_sampling() (in module torch_geometric.utils)
structured_negative_sampling_feasible() (in module torch_geometric.utils)
subgraph() (in module torch_geometric.utils)
SuiteSparseMatrixCollection (class in torch_geometric.datasets)
SuperGATConv (class in torch_geometric.nn.conv)
SVDFeatureReduction (class in torch_geometric.transforms)
T
TAGConv (class in torch_geometric.nn.conv)
TargetIndegree (class in torch_geometric.transforms)
test() (DeepGraphInfomax method)
(GAE method)
(MetaPath2Vec method)
(Node2Vec method)
(RENet method)
(SignedGCN method)
TGNMemory (class in torch_geometric.nn.models)
timeit() (in module torch_geometric.profile)
to() (Data method)
(HeteroData method)
to_cugraph() (in module torch_geometric.utils)
to_data_list() (Batch method)
to_dense_adj() (in module torch_geometric.utils)
to_dense_batch() (in module torch_geometric.utils)
to_dict() (Data method)
(HeteroData method)
to_hetero() (in module torch_geometric.nn.to_hetero_transformer)
to_hetero_with_bases() (in module torch_geometric.nn.to_hetero_with_bases_transformer)
to_heterogeneous() (Data method)
to_homogeneous() (HeteroData method)
to_namedtuple() (Data method)
(HeteroData method)
to_networkx() (in module torch_geometric.utils)
to_scipy_sparse_matrix() (in module torch_geometric.utils)
to_trimesh() (in module torch_geometric.utils)
to_undirected() (in module torch_geometric.utils)
ToDense (class in torch_geometric.transforms)
ToDevice (class in torch_geometric.transforms)
TopKPooling (class in torch_geometric.nn.pool)
torch_geometric.data
module
torch_geometric.datasets
module
torch_geometric.debug
module
torch_geometric.graphgym
module
torch_geometric.graphgym.models
module
torch_geometric.graphgym.utils
module
torch_geometric.loader
module
torch_geometric.nn.conv
module
torch_geometric.nn.conv.message_passing
module
torch_geometric.nn.data_parallel
module
torch_geometric.nn.functional
module
torch_geometric.nn.glob
module
torch_geometric.nn.meta
module
torch_geometric.nn.models
module
torch_geometric.nn.norm
module
torch_geometric.nn.pool
module
torch_geometric.nn.unpool
module
torch_geometric.profile
module
torch_geometric.seed
module
torch_geometric.transforms
module
torch_geometric.utils
module
TOSCA (class in torch_geometric.datasets)
ToSLIC (class in torch_geometric.transforms)
ToSparseTensor (class in torch_geometric.transforms)
ToUndirected (class in torch_geometric.transforms)
track_running_stats (InstanceNorm attribute)
TrackMLParticleTrackingDataset (class in torch_geometric.datasets)
train() (in module torch_geometric.graphgym)
(TGNMemory method)
train_test_split_edges() (in module torch_geometric.utils)
transform() (Transformer method)
Transformer (class in torch_geometric.nn.fx)
TransformerConv (class in torch_geometric.nn.conv)
transition_matrix() (GDC method)
tree_decomposition() (in module torch_geometric.utils)
triplets() (DimeNet method)
true_negative() (in module torch_geometric.utils)
true_positive() (in module torch_geometric.utils)
TUDataset (class in torch_geometric.datasets)
Twitch (class in torch_geometric.datasets)
TwoHop (class in torch_geometric.transforms)
U
unpool() (EdgePooling method)
unpool_description (EdgePooling attribute)
update() (MessagePassing method)
update_state() (TGNMemory method)
UPFD (class in torch_geometric.datasets)
url (DimeNet attribute)
(SchNet attribute)
V
VGAE (class in torch_geometric.nn.models)
visualize_subgraph() (GNNExplainer method)
voxel_grid() (in module torch_geometric.nn.pool)
W
WebKB (class in torch_geometric.datasets)
WikiCS (class in torch_geometric.datasets)
WikipediaNetwork (class in torch_geometric.datasets)
WILLOWObjectClass (class in torch_geometric.datasets)
WLConv (class in torch_geometric.nn.conv)
WordNet18 (class in torch_geometric.datasets)
WordNet18RR (class in torch_geometric.datasets)
X
XConv (class in torch_geometric.nn.conv)
Y
Yelp (class in torch_geometric.datasets)
Z
ZINC (class in torch_geometric.datasets)
Read the Docs
v: 2.0.2
Versions
latest
2.0.2
2.0.1
2.0.0
1.7.2
1.7.1
1.7.0
1.6.3
1.6.2
1.6.1
1.6.0
1.5.0
1.4.3
1.4.2
1.4.1
1.3.2
1.3.1
1.3.0
Downloads
On Read the Docs
Project Home
Builds