2.2.0
Notes
Installation
Introduction by Example
Creating Message Passing Networks
Creating Your Own Datasets
Heterogeneous Graph Learning
Loading Graphs from CSV
Managing Experiments with GraphGym
Advanced Mini-Batching
Memory-Efficient Aggregations
TorchScript Support
Scaling Up GNNs via Remote Backends
GNN Cheatsheet
Dataset Cheatsheet
Colab Notebooks and Video Tutorials
External Resources
Package Reference
torch_geometric
torch_geometric.nn
torch_geometric.data
torch_geometric.loader
torch_geometric.sampler
torch_geometric.datasets
torch_geometric.transforms
torch_geometric.utils
torch_geometric.graphgym
torch_geometric.explain
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
_
__call__() (Explainer method)
__cat_dim__() (Data method)
(HeteroData method)
(TemporalData method)
__inc__() (Data method)
(HeteroData method)
(TemporalData method)
A
Actor (class in torch_geometric.datasets)
add_random_edge() (in module torch_geometric.utils)
add_remaining_self_loops() (in module torch_geometric.utils)
add_self_loops() (in module torch_geometric.utils)
AddLaplacianEigenvectorPE (class in torch_geometric.transforms)
AddMetaPaths (class in torch_geometric.transforms)
AddRandomMetaPaths (class in torch_geometric.transforms)
AddRandomWalkPE (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)
Aggregation (class in torch_geometric.nn.aggr)
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)
(TemporalData method)
apply_() (Data method)
(HeteroData method)
(TemporalData method)
APPNP (class in torch_geometric.nn.conv)
AQSOL (class in torch_geometric.datasets)
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)
assortativity() (in module torch_geometric.utils)
AtomEncoder (class in torch_geometric.graphgym.models)
att_x (SuperGATConv attribute)
att_y (SuperGATConv attribute)
AttentionalAggregation (class in torch_geometric.nn.aggr)
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)
available_explanations (Explanation property)
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)
BaseSampler (class in torch_geometric.sampler.base)
BaseTransform (class in torch_geometric.transforms)
BAShapes (class in torch_geometric.datasets)
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)
bipartite_subgraph() (in module torch_geometric.utils)
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_global_pooling_module() (Transformer method)
call_message_passing_module() (Transformer method)
call_method() (Transformer method)
call_module() (Transformer method)
captum_output_to_dicts() (in module torch_geometric.nn.models)
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)
(TemporalData 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)
collate() (InMemoryDataset static method)
collate_fn() (LinkLoader method)
(NodeLoader 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)
connect() (ExplainerAlgorithm method)
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)
(TemporalData method)
coo() (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)
(TemporalData 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)
csc() (Data method)
(HeteroData method)
csr() (Data method)
(HeteroData method)
cuda() (Data method)
(HeteroData method)
(TemporalData 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)
DegreeScalerAggregation (class in torch_geometric.nn.aggr)
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)
(TemporalData method)
(TGNMemory method)
detach_() (Data method)
(HeteroData method)
(TemporalData method)
DGraphFin (class in torch_geometric.datasets)
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)
DimeNetPlusPlus (class in torch_geometric.nn.models)
discriminate() (DeepGraphInfomax method)
(SignedGCN method)
discriminator_loss() (ARGA method)
disjoint (NeighborSampler property)
Distance (class in torch_geometric.transforms)
DMoNPooling (class in torch_geometric.nn.dense)
DNAConv (class in torch_geometric.nn.conv)
download() (Dataset method)
download_url() (in module torch_geometric.data)
dropout_adj() (in module torch_geometric.utils)
dropout_edge() (in module torch_geometric.utils)
dropout_node() (in module torch_geometric.utils)
dropout_path() (in module torch_geometric.utils)
dummy_context (class in torch_geometric.graphgym.utils)
DummyExplainer (class in torch_geometric.explain.algorithm)
dump_cfg() (in module torch_geometric.graphgym)
DynamicBatchSampler (class in torch_geometric.loader)
DynamicEdgeConv (class in torch_geometric.nn.conv)
DynamicFAUST (class in torch_geometric.datasets)
E
ECConv (in module torch_geometric.nn.conv)
edge_items() (HeteroData method)
edge_permutation (BaseSampler property)
edge_stores (HeteroData property)
edge_type_subgraph() (HeteroData method)
edge_types (HeteroData property)
edge_update() (GATConv method)
(MessagePassing method)
edge_updater() (MessagePassing method)
EdgeCNN (class in torch_geometric.nn.models)
EdgeConv (class in torch_geometric.nn.conv)
EdgePooling (class in torch_geometric.nn.pool)
EGConv (class in torch_geometric.nn.conv)
EllipticBitcoinDataset (class in torch_geometric.datasets)
EmailEUCore (class in torch_geometric.datasets)
embed() (RECT_L method)
enable_cpu_affinity() (NodeLoader method)
encode() (ARGVA method)
(GAE method)
(VGAE method)
Entities (class in torch_geometric.datasets)
eps (InstanceNorm attribute)
EquilibriumAggregation (class in torch_geometric.nn.aggr)
erdos_renyi_graph() (in module torch_geometric.utils)
Explainer (class in torch_geometric.explain)
explainer_config (ExplainerAlgorithm property)
ExplainerAlgorithm (class in torch_geometric.explain.algorithm)
ExplainerConfig (class in torch_geometric.explain.config)
Explanation (class in torch_geometric.explain)
extract() (RootedEgoNets method)
(RootedRWSubgraph 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
FacebookPagePage (class in torch_geometric.datasets)
FaceToEdge (class in torch_geometric.transforms)
FAConv (class in torch_geometric.nn.conv)
FakeDataset (class in torch_geometric.datasets)
FakeHeteroDataset (class in torch_geometric.datasets)
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)
FeaturePropagation (class in torch_geometric.transforms)
FiLMConv (class in torch_geometric.nn.conv)
filter_fn() (LinkLoader method)
(NodeLoader method)
FixedPoints (class in torch_geometric.transforms)
Flickr (class in torch_geometric.datasets)
forward() (Aggregation method)
(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)
(DMoNPooling method)
(DNAConv method)
(DynamicEdgeConv method)
(EdgeConv method)
(EdgePooling method)
(EGConv method)
(ExplainerAlgorithm method)
(FAConv method)
(FastRGCNConv method)
(FeaStConv method)
(FiLMConv method)
(FusedGATConv method)
(GATConv method)
(GatedGraphConv method)
(GATv2Conv method)
(GCN2Conv method)
(GCNConv method)
(GENConv method)
(GeneralConv method)
(GINConv method)
(GINEConv method)
(GMMConv method)
(GraphConv method)
(GraphNorm method)
(GraphSizeNorm method)
(GraphUNet method)
(GravNetConv method)
(HANConv method)
(HEATConv method)
(HeteroConv method)
(HeteroLinear method)
(HGTConv method)
(HypergraphConv method)
(InnerProductDecoder method)
(InstanceNorm method)
(JumpingKnowledge method)
(LabelPropagation method)
(LayerNorm method)
(LEConv method)
(LGConv method)
(LightGCN method)
(Linear method)
(LINKX method)
(MaskLabel method)
(MeanSubtractionNorm method)
(MemPooling method)
(MessageNorm method)
(MetaLayer method)
(MetaPath2Vec method)
(MFConv method)
(MLP method)
(NNConv method)
(Node2Vec method)
(PairNorm method)
(PANConv method)
(PANPooling method)
(PDNConv method)
(PNAConv method)
(PointNetConv method)
(PointTransformerConv method)
(PositionalEncoding method)
(PPFConv method)
(RECT_L method)
(RENet method)
(ResGatedGraphConv method)
(RGATConv method)
(RGCNConv method)
(SAGEConv method)
(SAGPooling method)
(SchNet method)
(SGConv method)
(SignedConv method)
(SignedGCN method)
(SplineConv method)
(SSGConv method)
(SuperGATConv method)
(TAGConv method)
(TGNMemory method)
(TopKPooling method)
(TransformerConv method)
(WLConv method)
(WLConvContinuous 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 class method)
(DimeNetPlusPlus class method)
(SchNet static method)
from_scipy_sparse_matrix() (in module torch_geometric.utils)
from_smiles() (in module torch_geometric.utils)
from_trimesh() (in module torch_geometric.utils)
FusedGATConv (class in torch_geometric.nn.conv)
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_all_edge_attrs() (Data method)
(HeteroData method)
get_all_tensor_attrs() (Data method)
(HeteroData method)
get_attention() (SuperGATConv method)
get_attention_loss() (SuperGATConv method)
get_attr() (Transformer method)
get_complement_subgraph() (Explanation 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_degree_histogram() (PNAConv static method)
get_edge_index() (Data method)
(HeteroData method)
get_edge_store() (HeteroData method)
get_embedding() (LightGCN method)
get_example() (Batch method)
get_explanation_subgraph() (Explanation method)
get_fname() (in module torch_geometric.graphgym)
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_mesh_laplacian() (in module torch_geometric.utils)
get_model_size() (in module torch_geometric.profile)
get_node_store() (HeteroData method)
get_num_hops() (in module torch_geometric.utils)
get_prediction() (Explainer method)
get_self_loop_attr() (in module torch_geometric.utils)
get_semantic_labels() (RECT_L method)
get_stats_summary() (in module torch_geometric.profile)
get_summary() (Dataset method)
get_tensor() (Data method)
(HeteroData method)
get_tensor_size() (Data method)
(HeteroData method)
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.pool)
global_max_pool() (in module torch_geometric.graphgym.models)
(in module torch_geometric.nn.pool)
global_mean_pool() (in module torch_geometric.graphgym.models)
(in module torch_geometric.nn.pool)
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.explain.algorithm)
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.aggr)
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)
GroupAddRev (class in torch_geometric.nn.models)
H
HANConv (class in torch_geometric.nn.conv)
has_edge_level_arg() (Transformer method)
has_graph_level_arg() (Transformer method)
has_isolated_nodes() (Data method)
(HeteroData method)
has_node_level_arg() (Transformer 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)
HGBDataset (class in torch_geometric.datasets)
HGTConv (class in torch_geometric.nn.conv)
HGTLoader (class in torch_geometric.loader)
HGTSampler (class in torch_geometric.sampler)
histogram() (WLConv method)
homophily() (in module torch_geometric.utils)
HydroNet (class in torch_geometric.datasets)
HypergraphConv (class in torch_geometric.nn.conv)
I
ICEWS18 (class in torch_geometric.datasets)
ImbalancedSampler (class in torch_geometric.loader)
IMDB (class in torch_geometric.datasets)
in_channels (MLP property)
index_select() (Batch method)
(Dataset method)
index_to_mask() (in module torch_geometric.utils)
IndexToMask (class in torch_geometric.transforms)
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)
is_ckpt_epoch() (in module torch_geometric.graphgym.utils)
is_coalesced() (Data method)
(HeteroData method)
(TemporalData method)
is_cuda (Data property)
(HeteroData property)
(TemporalData property)
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_graph_level() (Transformer method)
is_node_attr() (Data method)
is_node_level() (Transformer method)
is_sparse() (in module torch_geometric.utils)
is_temporal (NeighborSampler property)
is_torch_sparse_tensor() (in module torch_geometric.utils)
is_undirected() (Data method)
(HeteroData method)
(in module torch_geometric.utils)
items() (Data method)
J
jittable() (ASAPooling method)
(AttentiveFP method)
(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)
(TemporalData 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)
LargestConnectedComponents (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)
(HydroNet method)
(InMemoryDataset method)
LGConv (class in torch_geometric.nn.conv)
LightGCN (class in torch_geometric.nn.models)
LightningDataset (class in torch_geometric.data)
LightningLinkData (class in torch_geometric.data)
LightningNodeData (class in torch_geometric.data)
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)
link_pred_loss() (LightGCN method)
LinkLoader (class in torch_geometric.loader)
LinkNeighborLoader (class in torch_geometric.loader)
LINKX (class in torch_geometric.nn.models)
LINKXDataset (class in torch_geometric.datasets)
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)
LRGBDataset (class in torch_geometric.datasets)
LSTMAggregation (class in torch_geometric.nn.aggr)
M
makedirs() (in module torch_geometric.data)
makedirs_rm_exist() (in module torch_geometric.graphgym.utils)
MalNetTiny (class in torch_geometric.datasets)
mask_feature() (in module torch_geometric.utils)
mask_to_index() (in module torch_geometric.utils)
MaskLabel (class in torch_geometric.nn.models)
MaskToIndex (class in torch_geometric.transforms)
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)
MaxAggregation (class in torch_geometric.nn.aggr)
MD17 (class in torch_geometric.datasets)
MeanAggregation (class in torch_geometric.nn.aggr)
MeanSubtractionNorm (class in torch_geometric.nn.norm)
MedianAggregation (class in torch_geometric.nn.aggr)
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)
metadata() (HeteroData method)
MetaLayer (class in torch_geometric.nn.meta)
MetaPath2Vec (class in torch_geometric.nn.models)
MFConv (class in torch_geometric.nn.conv)
MinAggregation (class in torch_geometric.nn.aggr)
MixHopSyntheticDataset (class in torch_geometric.datasets)
MLP (class in torch_geometric.graphgym.models)
(class in torch_geometric.nn.models)
MNISTSuperpixels (class in torch_geometric.datasets)
model_config (ExplainerAlgorithm property)
ModelConfig (class in torch_geometric.explain.config)
ModelNet (class in torch_geometric.datasets)
module
torch_geometric.data
torch_geometric.datasets
torch_geometric.debug
torch_geometric.explain.algorithm
torch_geometric.graphgym
torch_geometric.graphgym.models
torch_geometric.graphgym.utils
torch_geometric.loader
torch_geometric.nn.aggr
torch_geometric.nn.conv
torch_geometric.nn.data_parallel
torch_geometric.nn.encoding
torch_geometric.nn.functional
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.sampler
torch_geometric.seed
torch_geometric.transforms
torch_geometric.utils
MoleculeNet (class in torch_geometric.datasets)
momentum (InstanceNorm attribute)
MovieLens (class in torch_geometric.datasets)
MulAggregation (class in torch_geometric.nn.aggr)
multi_get_tensor() (Data method)
(HeteroData method)
MultiAggregation (class in torch_geometric.nn.aggr)
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)
(class in torch_geometric.sampler)
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_items() (HeteroData method)
node_stores (HeteroData property)
node_type_subgraph() (HeteroData method)
node_types (HeteroData property)
NodeLoader (class in torch_geometric.loader)
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 (Dataset property)
(InMemoryDataset property)
num_edge_features (Data property)
(Dataset property)
(HeteroData property)
num_edges (Data property)
(HeteroData property)
(TemporalData property)
num_events (TemporalData property)
num_faces (Data property)
num_features (Data property)
(Dataset property)
(HeteroData property)
(InstanceNorm attribute)
num_graphs (Batch property)
num_groups (GroupAddRev property)
num_layers (MLP property)
num_node_features (Data property)
(Dataset property)
(HeteroData property)
num_nodes (Data property)
(HeteroData property)
(TemporalData property)
O
OGB_MAG (class in torch_geometric.datasets)
OMDB (class in torch_geometric.datasets)
OneHotDegree (class in torch_geometric.transforms)
out_channels (MLP property)
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)
(TemporalData 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)
PolBlogs (class in torch_geometric.datasets)
pos_embedding_loss() (SignedGCN method)
pos_sample() (Node2Vec method)
PositionalEncoding (class in torch_geometric.nn.encoding)
positive_sampling() (SuperGATConv method)
PowerMeanAggregation (class in torch_geometric.nn.aggr)
PPFConv (class in torch_geometric.nn.conv)
PPI (class in torch_geometric.datasets)
pre_transform() (RENet static method)
predict_link() (LightGCN method)
print_summary() (Dataset method)
process() (Dataset method)
processed_file_names (Dataset property)
(InMemoryDataset property)
processed_paths (Dataset property)
profileit() (in module torch_geometric.profile)
propagate() (MessagePassing method)
put_edge_index() (Data method)
(HeteroData method)
put_tensor() (Data method)
(HeteroData method)
Q
QM7b (class in torch_geometric.datasets)
QM9 (class in torch_geometric.datasets)
QuantileAggregation (class in torch_geometric.nn.aggr)
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)
RandomJitter (class in torch_geometric.transforms)
RandomLinkSplit (class in torch_geometric.transforms)
RandomNodeLoader (class in torch_geometric.loader)
RandomNodeSplit (class in torch_geometric.transforms)
RandomPartitionGraphDataset (class in torch_geometric.datasets)
RandomRotate (class in torch_geometric.transforms)
RandomScale (class in torch_geometric.transforms)
RandomShear (class in torch_geometric.transforms)
ratio_mask() (MaskLabel static method)
raw_file_names (Dataset property)
(InMemoryDataset property)
raw_paths (Dataset property)
recommend() (LightGCN method)
recommendation_loss() (LightGCN method)
recon_loss() (GAE method)
record_stream() (Data method)
(HeteroData method)
(TemporalData 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_dataset() (in module torch_geometric.graphgym)
register_edge_encoder() (in module torch_geometric.graphgym)
register_edge_update_forward_hook() (MessagePassing method)
register_edge_update_forward_pre_hook() (MessagePassing method)
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_metric() (in module torch_geometric.graphgym)
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_ckpt() (in module torch_geometric.graphgym)
remove_isolated_nodes() (in module torch_geometric.utils)
remove_self_loops() (in module torch_geometric.utils)
remove_tensor() (Data method)
(HeteroData method)
RemoveIsolatedNodes (class in torch_geometric.transforms)
RemoveTrainingClasses (class in torch_geometric.transforms)
rename() (HeteroData method)
rename_profile_file() (in module torch_geometric.profile)
RENet (class in torch_geometric.nn.models)
reparametrize() (ARGVA method)
(VGAE method)
replace_all_uses_with() (Transformer method)
requires_grad_() (Data method)
(HeteroData method)
(TemporalData method)
reset() (timeit 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)
(DMoNPooling 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)
(GMMConv method)
(GraphConv method)
(GraphNorm method)
(GraphUNet method)
(GravNetConv method)
(GroupAddRev method)
(HANConv method)
(HEATConv method)
(HeteroConv method)
(HeteroLinear method)
(HGTConv method)
(HypergraphConv method)
(JumpingKnowledge method)
(LayerNorm method)
(LEConv method)
(LGConv method)
(LightGCN method)
(Linear method)
(LINKX method)
(MaskLabel method)
(MeanSubtractionNorm method)
(MemPooling method)
(MessageNorm method)
(MetaLayer method)
(MetaPath2Vec method)
(MFConv method)
(MLP 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)
(RGATConv method)
(RGCNConv method)
(SAGEConv method)
(SAGPooling method)
(SchNet method)
(SGConv method)
(SignedConv method)
(SignedGCN method)
(SplineConv method)
(SSGConv 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)
RGATConv (class in torch_geometric.nn.conv)
RGCNConv (class in torch_geometric.nn.conv)
RootedEgoNets (class in torch_geometric.transforms)
RootedRWSubgraph (class in torch_geometric.transforms)
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() (AddRandomMetaPaths static method)
(Node2Vec method)
sample_adj() (AddMetaPaths method)
sample_from_edges() (BaseSampler method)
sample_from_nodes() (BaseSampler 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.aggr)
set_cfg() (in module torch_geometric.graphgym)
set_debug (class in torch_geometric.debug)
set_out_dir() (in module torch_geometric.graphgym)
set_printing() (in module torch_geometric.graphgym)
set_run_dir() (in module torch_geometric.graphgym)
set_validate_args() (Aggregation static method)
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)
(TemporalData method)
SHREC2016 (class in torch_geometric.datasets)
shuffle() (Dataset method)
shuffle_node() (in module torch_geometric.utils)
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)
(TemporalData method)
smooth() (CorrectAndSmooth method)
SNAPDataset (class in torch_geometric.datasets)
softmax() (in module torch_geometric.utils)
SoftmaxAggregation (class in torch_geometric.nn.aggr)
sort_edge_index() (in module torch_geometric.utils)
SortAggregation (class in torch_geometric.nn.aggr)
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)
spmm() (in module torch_geometric.utils)
SSGConv (class in torch_geometric.nn.conv)
StdAggregation (class in torch_geometric.nn.aggr)
stochastic_blockmodel_graph() (in module torch_geometric.utils)
StochasticBlockModelDataset (class in torch_geometric.datasets)
stores (HeteroData property)
structured_negative_sampling() (in module torch_geometric.utils)
structured_negative_sampling_feasible() (in module torch_geometric.utils)
subgraph() (Data method)
(HeteroData method)
(in module torch_geometric.utils)
SuiteSparseMatrixCollection (class in torch_geometric.datasets)
SumAggregation (class in torch_geometric.nn.aggr)
SuperGATConv (class in torch_geometric.nn.conv)
supports() (ExplainerAlgorithm method)
supports_edge_attr (EdgeCNN attribute)
(GAT attribute)
(GCN attribute)
(GIN attribute)
(GraphSAGE attribute)
(PNA attribute)
supports_edge_weight (EdgeCNN attribute)
(GAT attribute)
(GCN attribute)
(GIN attribute)
(GraphSAGE attribute)
(PNA attribute)
SVDFeatureReduction (class in torch_geometric.transforms)
T
TAGConv (class in torch_geometric.nn.conv)
TargetIndegree (class in torch_geometric.transforms)
TemporalData (class in torch_geometric.data)
TemporalDataLoader (class in torch_geometric.loader)
test() (DeepGraphInfomax method)
(GAE method)
(MetaPath2Vec method)
(Node2Vec method)
(RENet method)
(SignedGCN method)
TGNMemory (class in torch_geometric.nn.models)
ThresholdConfig (class in torch_geometric.explain.config)
timeit (class in torch_geometric.profile)
to() (Data method)
(HeteroData method)
(TemporalData method)
to_captum() (in module torch_geometric.nn.models)
to_captum_input() (in module torch_geometric.nn.models)
to_captum_model() (in module torch_geometric.nn.models)
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)
(TemporalData method)
to_graph_format() (FusedGATConv static 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)
(TemporalData method)
to_networkx() (in module torch_geometric.utils)
to_scipy_sparse_matrix() (in module torch_geometric.utils)
to_smiles() (in module torch_geometric.utils)
to_torch_coo_tensor() (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.explain.algorithm
module
torch_geometric.graphgym
module
torch_geometric.graphgym.models
module
torch_geometric.graphgym.utils
module
torch_geometric.loader
module
torch_geometric.nn.aggr
module
torch_geometric.nn.conv
module
torch_geometric.nn.data_parallel
module
torch_geometric.nn.encoding
module
torch_geometric.nn.functional
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.sampler
module
torch_geometric.seed
module
torch_geometric.transforms
module
torch_geometric.utils
module
torch_profile() (in module torch_geometric.profile)
TOSCA (class in torch_geometric.datasets)
ToSLIC (class in torch_geometric.transforms)
ToSparseTensor (class in torch_geometric.transforms)
ToUndirected (class in torch_geometric.transforms)
trace_handler() (in module torch_geometric.profile)
track_running_stats (InstanceNorm attribute)
train() (TGNMemory method)
train_test_split_edges() (in module torch_geometric.utils)
train_val_test_split() (TemporalData method)
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)
TUDataset (class in torch_geometric.datasets)
Twitch (class in torch_geometric.datasets)
TwoHop (class in torch_geometric.transforms)
U
unbatch() (in module torch_geometric.utils)
unbatch_edge_index() (in module torch_geometric.utils)
unpool() (EdgePooling method)
update() (MessagePassing method)
update_state() (TGNMemory method)
update_tensor() (Data method)
(HeteroData method)
UPFD (class in torch_geometric.datasets)
url (DimeNet attribute)
(DimeNetPlusPlus attribute)
(SchNet attribute)
V
validate() (Data method)
(Explanation method)
(HeteroData method)
VarAggregation (class in torch_geometric.nn.aggr)
VGAE (class in torch_geometric.nn.models)
view() (Data method)
(HeteroData method)
VirtualNode (class in torch_geometric.transforms)
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)
WLConvContinuous (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.2.0
Versions
latest
2.2.0
2.1.0
2.0.4
2.0.3
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