latest
Install PyG
Installation
Get Started
Introduction by Example
Colab Notebooks and Video Tutorials
Tutorials
Design of Graph Neural Networks
Working with Graph Datasets
Use-Cases & Applications
Distributed Training
Advanced Concepts
Advanced Mini-Batching
Memory-Efficient Aggregations
Hierarchical Neighborhood Sampling
Compiled Graph Neural Networks
TorchScript Support
Scaling Up GNNs via Remote Backends
Managing Experiments with GraphGym
CPU Affinity for PyG Workloads
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.explain
torch_geometric.metrics
torch_geometric.distributed
torch_geometric.contrib
torch_geometric.graphgym
torch_geometric.profile
Cheatsheets
GNN Cheatsheet
Dataset Cheatsheet
External Resources
External Resources
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() (KNNIndex method)
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)
AddRemainingSelfLoops (class in torch_geometric.transforms)
AddSelfLoops (class in torch_geometric.transforms)
AffinityMixin (class in torch_geometric.loader)
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)
AirfRANS (class in torch_geometric.datasets)
Airports (class in torch_geometric.datasets)
Amazon (class in torch_geometric.datasets)
AmazonBook (class in torch_geometric.datasets)
AmazonProducts (class in torch_geometric.datasets)
AMiner (class in torch_geometric.datasets)
AntiSymmetricConv (class in torch_geometric.nn.conv)
append() (OnDiskDataset method)
apply() (Data method)
(HeteroData method)
(TemporalData method)
apply_() (Data method)
(HeteroData method)
(TemporalData method)
APPNP (class in torch_geometric.nn.conv)
approx_knn() (in module torch_geometric.nn.pool)
approx_knn_graph() (in module torch_geometric.nn.pool)
ApproxL2KNNIndex (class in torch_geometric.nn.pool)
ApproxMIPSKNNIndex (class in torch_geometric.nn.pool)
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)
as_tensor() (EdgeIndex method)
ASAPooling (class in torch_geometric.nn.pool)
assortativity() (in module torch_geometric.utils)
AtomEncoder (class in torch_geometric.graphgym.models)
attack() (PRBCDAttack method)
AttentionalAggregation (class in torch_geometric.nn.aggr)
AttentionExplainer (class in torch_geometric.explain.algorithm)
AttentiveFP (class in torch_geometric.nn.models)
AttributedGraphDataset (class in torch_geometric.datasets)
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
BA2MotifDataset (class in torch_geometric.datasets)
BAGraph (class in torch_geometric.datasets.graph_generator)
BAMultiShapesDataset (class in torch_geometric.datasets)
barabasi_albert_graph() (in module torch_geometric.utils)
BaseSampler (class in torch_geometric.sampler)
BaseTransform (class in torch_geometric.transforms)
BAShapes (class in torch_geometric.datasets)
Batch (class in torch_geometric.data)
batch_size (Batch property)
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)
benchmark() (in module torch_geometric.profile)
bipartite_subgraph() (in module torch_geometric.utils)
BitcoinOTC (class in torch_geometric.datasets)
BondEncoder (class in torch_geometric.graphgym.models)
BrcaTcga (class in torch_geometric.datasets)
bro() (in module torch_geometric.nn.functional)
C
CachedLoader (class in torch_geometric.loader)
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 (class in torch_geometric.nn.models)
CaptumExplainer (class in torch_geometric.explain.algorithm)
Cartesian (class in torch_geometric.transforms)
Center (class in torch_geometric.transforms)
CGConv (class in torch_geometric.nn.conv)
characterization_score() (in module torch_geometric.explain.metric)
ChebConv (class in torch_geometric.nn.conv)
CitationFull (class in torch_geometric.datasets)
clean_ckpt() (in module torch_geometric.graphgym)
clear() (CachedLoader method)
clone() (Data method)
(HeteroData method)
(TemporalData method)
close() (Database method)
(OnDiskDataset method)
(RocksDatabase method)
(SQLiteDatabase 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)
(TemporalData method)
Coauthor (class in torch_geometric.datasets)
coeffs (GRBCDAttack attribute)
(PRBCDAttack attribute)
collate() (InMemoryDataset static method)
collate_fn() (LinkLoader method)
(NodeLoader method)
collect() (HeteroData method)
CoMA (class in torch_geometric.datasets)
ComplEx (class in torch_geometric.nn.kge)
Compose (class in torch_geometric.transforms)
ComposeFilters (class in torch_geometric.transforms)
compute() (LinkPredF1 method)
(LinkPredMAP method)
(LinkPredNDCG method)
(LinkPredPrecision method)
(LinkPredRecall method)
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)
concat() (Data method)
(HeteroData method)
(TemporalData method)
connect() (Database method)
(ExplainerAlgorithm method)
(RocksDatabase method)
(SQLiteDatabase 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)
(GraphStore 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)
(InMemoryDataset 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)
(GraphStore method)
(HeteroData method)
csr() (Data method)
(GraphStore method)
(HeteroData method)
cuda() (Data method)
(HeteroData method)
(InMemoryDataset method)
(TemporalData method)
CuGraphGATConv (class in torch_geometric.nn.conv)
CuGraphRGCNConv (class in torch_geometric.nn.conv)
CuGraphSAGEConv (class in torch_geometric.nn.conv)
cumsum() (in module torch_geometric.utils)
CustomMotif (class in torch_geometric.datasets.motif_generator)
CycleMotif (class in torch_geometric.datasets.motif_generator)
D
Data (class in torch_geometric.data)
Database (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)
db (OnDiskDataset property)
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)
DeepSetsAggregation (class in torch_geometric.nn.aggr)
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() (in module torch_geometric.nn.dense)
dense_mincut_pool() (in module torch_geometric.nn.dense)
dense_to_sparse() (in module torch_geometric.utils)
DenseDataLoader (class in torch_geometric.loader)
DenseGATConv (class in torch_geometric.nn.dense)
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)
deserialize() (OnDiskDataset method)
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)
DimeNet (class in torch_geometric.nn.models)
DimeNetPlusPlus (class in torch_geometric.nn.models)
DirGNNConv (class in torch_geometric.nn.conv)
disable_dynamic_shapes() (in module torch_geometric.experimental)
discriminate() (DeepGraphInfomax method)
(SignedGCN method)
discriminator_loss() (ARGA method)
Distance (class in torch_geometric.transforms)
DistContext (class in torch_geometric.distributed)
DistLinkNeighborLoader (class in torch_geometric.distributed)
DistLoader (class in torch_geometric.distributed)
DistMult (class in torch_geometric.nn.kge)
DistNeighborLoader (class in torch_geometric.distributed)
DistNeighborSampler (class in torch_geometric.distributed)
DMoNPooling (class in torch_geometric.nn.dense)
DNAConv (class in torch_geometric.nn.conv)
download() (Dataset method)
(InMemoryDataset method)
(OnDiskDataset method)
download_google_url() (in module torch_geometric.data)
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
edge_attrs() (Data method)
(HeteroData method)
(TemporalData method)
edge_index (TemporalData property)
edge_items() (HeteroData method)
edge_permutation (BaseSampler property)
edge_sample() (DistNeighborSampler method)
edge_stores (HeteroData property)
edge_subgraph() (Data method)
(HeteroData method)
edge_type_subgraph() (HeteroData method)
edge_types (HeteroData property)
edge_update() (MessagePassing method)
edge_updater() (MessagePassing method)
EdgeAttr (class in torch_geometric.data)
EdgeCNN (class in torch_geometric.nn.models)
EdgeConv (class in torch_geometric.nn.conv)
EdgeIndex (class in torch_geometric.edge_index)
EdgePooling (class in torch_geometric.nn.pool)
EdgeSamplerInput (class in torch_geometric.sampler)
EGConv (class in torch_geometric.nn.conv)
EllipticBitcoinDataset (class in torch_geometric.datasets)
EllipticBitcoinTemporalDataset (class in torch_geometric.datasets)
EmailEUCore (class in torch_geometric.datasets)
enable_cpu_affinity() (AffinityMixin method)
encode() (GAE method)
Entities (class in torch_geometric.datasets)
EquilibriumAggregation (class in torch_geometric.nn.aggr)
erdos_renyi_graph() (in module torch_geometric.utils)
ERGraph (class in torch_geometric.datasets.graph_generator)
experimental_mode (class in torch_geometric.experimental)
Explainer (class in torch_geometric.explain)
explainer_config (ExplainerAlgorithm property)
ExplainerAlgorithm (class in torch_geometric.explain.algorithm)
ExplainerConfig (class in torch_geometric.explain.config)
ExplainerDataset (class in torch_geometric.datasets)
Explanation (class in torch_geometric.explain)
extend() (OnDiskDataset 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)
FB15k_237 (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)
FeatureStore (class in torch_geometric.data)
fidelity() (in module torch_geometric.explain.metric)
fidelity_curve_auc() (in module torch_geometric.explain.metric)
fill_cache_() (EdgeIndex method)
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)
(AntiSymmetricConv method)
(APPNP method)
(ARGA method)
(ARGVA method)
(ARMAConv method)
(ASAPooling method)
(AttentionalAggregation method)
(AttentionExplainer method)
(AttentiveFP method)
(BatchNorm method)
(CaptumExplainer method)
(CGConv method)
(ChebConv method)
(ClusterGCNConv method)
(ComplEx method)
(CorrectAndSmooth method)
(CuGraphGATConv method)
(CuGraphRGCNConv method)
(CuGraphSAGEConv method)
(DeepGCNLayer method)
(DeepGraphInfomax method)
(DeepSetsAggregation method)
(DegreeScalerAggregation method)
(DenseGATConv method)
(DenseGCNConv method)
(DenseGINConv method)
(DenseGraphConv method)
(DenseSAGEConv method)
(DiffGroupNorm method)
(DimeNet method)
(DimeNetPlusPlus method)
(DirGNNConv method)
(DistMult method)
(DMoNPooling method)
(DNAConv method)
(DummyExplainer method)
(DynamicEdgeConv method)
(EdgeCNN method)
(EdgeConv method)
(EdgePooling method)
(EGConv method)
(EquilibriumAggregation method)
(ExplainerAlgorithm method)
(FAConv method)
(FastRGCNConv method)
(FeaStConv method)
(FiLMConv method)
(FusedGATConv method)
(GAE method)
(GAT method)
(GATConv method)
(GatedGraphConv method)
(GATv2Conv method)
(GCN method)
(GCN2Conv method)
(GCNConv method)
(GENConv method)
(GeneralConv method)
(GIN method)
(GINConv method)
(GINEConv method)
(GMMConv method)
(GNNExplainer method)
(GNNFF method)
(GPSConv method)
(GraphConv method)
(GraphMaskExplainer method)
(GraphMultisetTransformer method)
(GraphNorm method)
(GraphSAGE method)
(GraphSizeNorm method)
(GraphUNet method)
(GravNetConv method)
(GroupAddRev method)
(GRUAggregation method)
(HANConv method)
(HEATConv method)
(HeteroBatchNorm method)
(HeteroConv method)
(HeteroDictLinear method)
(HeteroLayerNorm method)
(HeteroLinear method)
(HGTConv method)
(HypergraphConv method)
(InnerProductDecoder method)
(InstanceNorm method)
(JumpingKnowledge method)
(KGEModel method)
(LabelPropagation method)
(LayerNorm method)
(LCMAggregation method)
(LEConv method)
(LGConv method)
(LightGCN method)
(Linear method)
(LINKX method)
(LSTMAggregation method)
(MaskLabel method)
(MaxAggregation method)
(MeanAggregation method)
(MeanSubtractionNorm method)
(MemPooling method)
(MessageNorm method)
(MessagePassing method)
(MetaLayer method)
(MetaPath2Vec method)
(MFConv method)
(MinAggregation method)
(MixHopConv method)
(MLP method)
(MLPAggregation method)
(MulAggregation method)
(MultiAggregation method)
(NeuralFingerprint method)
(NNConv method)
(Node2Vec method)
(PairNorm method)
(PANConv method)
(PANPooling method)
(PDNConv method)
(PGExplainer method)
(PGMExplainer method)
(PMLP method)
(PNA method)
(PNAConv method)
(PointGNNConv method)
(PointNetConv method)
(PointTransformerConv method)
(PositionalEncoding method)
(PowerMeanAggregation method)
(PPFConv method)
(QuantileAggregation method)
(RECT_L method)
(RENet method)
(ResGatedGraphConv method)
(RGATConv method)
(RGCNConv method)
(RotatE method)
(SAGEConv method)
(SAGPooling method)
(SchNet method)
(Set2Set method)
(SetTransformerAggregation method)
(SGConv method)
(SignedConv method)
(SignedGCN method)
(SimpleConv method)
(SoftmaxAggregation method)
(SortAggregation method)
(SplineConv method)
(SSGConv method)
(StdAggregation method)
(SumAggregation method)
(SuperGATConv method)
(TAGConv method)
(TemporalEncoding method)
(TGNMemory method)
(TopKPooling method)
(TransE method)
(TransformerConv method)
(VarAggregation method)
(VariancePreservingAggregation method)
(VGAE method)
(ViSNet method)
(WLConv method)
(WLConvContinuous method)
(XConv method)
forward_all() (InnerProductDecoder method)
fps() (in module torch_geometric.nn.pool)
from_cugraph() (in module torch_geometric.utils)
from_data() (LocalFeatureStore class method)
(LocalGraphStore class method)
from_data_list() (Batch class method)
from_dgl() (in module torch_geometric.utils)
from_dict() (Data class method)
(HeteroData class method)
(TemporalData class method)
from_hetero_data() (LocalFeatureStore class method)
(LocalGraphStore class method)
from_nested_tensor() (in module torch_geometric.utils)
from_networkit() (in module torch_geometric.utils)
from_networkx() (in module torch_geometric.utils)
from_pretrained() (PyGModelHubMixin class method)
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)
fully_specify() (TensorAttr method)
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)
GDELTLite (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)
generate_ids() (Data method)
(HeteroData method)
(TemporalData method)
generate_partition() (Partitioner method)
GenerateMeshNormals (class in torch_geometric.transforms)
geodesic_distance() (in module torch_geometric.utils)
GeometricShapes (class in torch_geometric.datasets)
get() (Database method)
(Dataset method)
(InMemoryDataset method)
(OnDiskDataset method)
(RocksDatabase method)
(SQLiteDatabase method)
get_all_edge_attrs() (Data method)
(GraphStore method)
(HeteroData method)
(LocalGraphStore method)
get_all_tensor_attrs() (Data method)
(FeatureStore method)
(HeteroData method)
(LocalFeatureStore method)
get_attention_loss() (SuperGATConv method)
get_attr() (Transformer method)
get_complement_subgraph() (Explanation method)
(HeteroExplanation method)
get_cpu_memory_from_gc() (in module torch_geometric.profile)
get_csc() (EdgeIndex method)
get_csr() (EdgeIndex method)
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)
(GraphStore method)
(HeteroData method)
get_edge_store() (HeteroData method)
get_emb() (KNNIndex method)
get_embedding() (LightGCN method)
get_embeddings() (in module torch_geometric.utils)
get_example() (Batch method)
get_explanation_subgraph() (Explanation method)
(HeteroExplanation method)
get_fname() (in module torch_geometric.graphgym)
get_gpu_memory_from_gc() (in module torch_geometric.profile)
get_gpu_memory_from_ipex() (in module torch_geometric.profile)
get_gpu_memory_from_nvidia_smi() (in module torch_geometric.profile)
get_home_dir() (in module torch_geometric.home)
get_indptr() (EdgeIndex method)
get_laplacian() (in module torch_geometric.utils)
get_mapped_values() (NumNeighbors method)
get_masked_prediction() (Explainer method)
get_mesh_laplacian() (in module torch_geometric.utils)
get_model_size() (in module torch_geometric.profile)
get_node_store() (HeteroData method)
get_num_cols() (EdgeIndex method)
get_num_hops() (in module torch_geometric.utils)
get_num_rows() (EdgeIndex method)
get_partition_ids_from_eids() (LocalGraphStore method)
get_partition_ids_from_nids() (LocalGraphStore method)
get_ppr() (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_sparse_size() (EdgeIndex method)
get_stats_summary() (in module torch_geometric.profile)
get_summary() (Dataset method)
(InMemoryDataset method)
(OnDiskDataset method)
get_target() (Explainer method)
get_tensor() (Data method)
(FeatureStore method)
(HeteroData method)
get_tensor_size() (Data method)
(FeatureStore method)
(HeteroData method)
get_values() (NumNeighbors 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)
GNNFF (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)
GPSConv (class in torch_geometric.nn.conv)
graclus() (in module torch_geometric.nn.pool)
GraphConv (class in torch_geometric.nn.conv)
GraphGenerator (class in torch_geometric.datasets.graph_generator)
GraphMaskExplainer (class in torch_geometric.explain.algorithm)
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)
GraphStore (class in torch_geometric.data)
GraphUNet (class in torch_geometric.nn.models)
GravNetConv (class in torch_geometric.nn.conv)
GRBCDAttack (class in torch_geometric.contrib.nn.models)
grid() (in module torch_geometric.utils)
GridGraph (class in torch_geometric.datasets.graph_generator)
GridMotif (class in torch_geometric.datasets.motif_generator)
GridSampling (class in torch_geometric.transforms)
groundtruth_metrics() (in module torch_geometric.explain.metric)
group_argsort() (in module torch_geometric.utils)
group_cat() (in module torch_geometric.utils)
group_distance_ratio() (DiffGroupNorm static method)
GroupAddRev (class in torch_geometric.nn.models)
GRUAggregation (class in torch_geometric.nn.aggr)
H
HalfHop (class in torch_geometric.transforms)
HANConv (class in torch_geometric.nn.conv)
has_download (Dataset property)
(InMemoryDataset property)
(OnDiskDataset property)
has_edge_level_arg() (Transformer method)
has_graph_level_arg() (Transformer method)
has_isolated_nodes() (Data method)
(HeteroData method)
(TemporalData method)
has_node_level_arg() (Transformer method)
has_process (Dataset property)
(InMemoryDataset property)
(OnDiskDataset property)
has_self_loops() (Data method)
(HeteroData method)
(TemporalData method)
HEATConv (class in torch_geometric.nn.conv)
HeteroBatchNorm (class in torch_geometric.nn.norm)
HeteroConv (class in torch_geometric.nn.conv)
HeteroData (class in torch_geometric.data)
HeteroDictLinear (class in torch_geometric.nn.dense)
HeteroExplanation (class in torch_geometric.explain)
HeteroLayerNorm (class in torch_geometric.nn.norm)
HeteroLinear (class in torch_geometric.nn.dense)
HeterophilousGraphDataset (class in torch_geometric.datasets)
HeteroSamplerOutput (class in torch_geometric.sampler)
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)
HM (class in torch_geometric.datasets)
homophily() (in module torch_geometric.utils)
HouseMotif (class in torch_geometric.datasets.motif_generator)
HydroNet (class in torch_geometric.datasets)
HypergraphConv (class in torch_geometric.nn.conv)
I
ICEWS18 (class in torch_geometric.datasets)
IGMCDataset (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)
(InMemoryDataset method)
(OnDiskDataset method)
index_sort() (in module torch_geometric.utils)
index_to_mask() (in module torch_geometric.utils)
IndexToMask (class in torch_geometric.transforms)
InfectionDataset (class in torch_geometric.datasets)
init_submodule() (Transformer method)
init_weights() (in module torch_geometric.graphgym)
InMemoryDataset (class in torch_geometric.data)
InnerProductDecoder (class in torch_geometric.nn.models)
insert() (Database method)
(RocksDatabase method)
(SQLiteDatabase method)
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)
(TemporalData method)
is_edge_attr() (Data method)
is_edge_level() (Transformer method)
is_eval_epoch() (in module torch_geometric.graphgym.utils)
is_experimental_mode_enabled() (in module torch_geometric.experimental)
is_fully_specified() (TensorAttr method)
is_graph_level() (Transformer method)
is_node_attr() (Data method)
is_node_level() (Transformer method)
is_set() (TensorAttr method)
is_sorted (EdgeIndex property)
is_sorted() (Data method)
(HeteroData method)
(TemporalData method)
is_sorted_by_col (EdgeIndex property)
is_sorted_by_row (EdgeIndex property)
is_sorted_by_time() (Data method)
(HeteroData method)
(TemporalData method)
is_sparse() (in module torch_geometric.utils)
is_torch_sparse_tensor() (in module torch_geometric.utils)
is_undirected (EdgeIndex property)
is_undirected() (Data method)
(HeteroData method)
(in module torch_geometric.utils)
(TemporalData method)
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 method)
(HeteroData method)
(TemporalData method)
KGEModel (class in torch_geometric.nn.kge)
kl_loss() (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)
KNNIndex (class in torch_geometric.nn.pool)
L
L2KNNIndex (class in torch_geometric.nn.pool)
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)
LCMAggregation (class in torch_geometric.nn.aggr)
LEConv (class in torch_geometric.nn.conv)
len() (Dataset method)
(InMemoryDataset method)
(OnDiskDataset method)
lexsort() (in module torch_geometric.utils)
LGConv (class in torch_geometric.nn.conv)
LightGCN (class in torch_geometric.nn.models)
LightningDataset (class in torch_geometric.data.lightning)
LightningLinkData (class in torch_geometric.data.lightning)
LightningNodeData (class in torch_geometric.data.lightning)
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)
LinkPredF1 (class in torch_geometric.metrics)
LinkPredMAP (class in torch_geometric.metrics)
LinkPredNDCG (class in torch_geometric.metrics)
LinkPredPrecision (class in torch_geometric.metrics)
LinkPredRecall (class in torch_geometric.metrics)
LINKX (class in torch_geometric.nn.models)
LINKXDataset (class in torch_geometric.datasets)
load() (InMemoryDataset method)
load_cfg() (in module torch_geometric.graphgym)
load_ckpt() (in module torch_geometric.graphgym)
loader() (KGEModel method)
(MetaPath2Vec method)
LocalCartesian (class in torch_geometric.transforms)
LocalDegreeProfile (class in torch_geometric.transforms)
LocalFeatureStore (class in torch_geometric.distributed)
LocalGraphStore (class in torch_geometric.distributed)
lookup_features() (LocalFeatureStore method)
loss() (ComplEx method)
(DeepGraphInfomax method)
(DistMult method)
(KGEModel method)
(MetaPath2Vec method)
(Node2Vec method)
(RotatE method)
(SignedGCN method)
(TransE 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_select() (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)
matmul() (EdgeIndex method)
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)
MessageNorm (class in torch_geometric.nn.norm)
MessagePassing (class in torch_geometric.nn.conv)
metadata() (HeteroData method)
MetaLayer (class in torch_geometric.nn.models)
MetaPath2Vec (class in torch_geometric.nn.models)
MFConv (class in torch_geometric.nn.conv)
MinAggregation (class in torch_geometric.nn.aggr)
MIPSKNNIndex (class in torch_geometric.nn.pool)
MixHopConv (class in torch_geometric.nn.conv)
MixHopSyntheticDataset (class in torch_geometric.datasets)
MLP (class in torch_geometric.graphgym.models)
(class in torch_geometric.nn.models)
MLPAggregation (class in torch_geometric.nn.aggr)
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.compile
torch_geometric.contrib.datasets
torch_geometric.contrib.explain
torch_geometric.contrib.nn.conv
torch_geometric.contrib.nn.models
torch_geometric.contrib.transforms
torch_geometric.debug
torch_geometric.distributed
torch_geometric.experimental
torch_geometric.graphgym
torch_geometric.graphgym.models
torch_geometric.graphgym.utils
torch_geometric.home
torch_geometric.loader
torch_geometric.nn.data_parallel
torch_geometric.nn.encoding
torch_geometric.nn.model_hub
torch_geometric.nn.summary
torch_geometric.profile
torch_geometric.sampler
torch_geometric.seed
torch_geometric.utils
MoleculeNet (class in torch_geometric.datasets)
MotifGenerator (class in torch_geometric.datasets.motif_generator)
MovieLens (class in torch_geometric.datasets)
MovieLens100K (class in torch_geometric.datasets)
MovieLens1M (class in torch_geometric.datasets)
MulAggregation (class in torch_geometric.nn.aggr)
multi_get() (Database method)
(OnDiskDataset method)
(RocksDatabase method)
(SQLiteDatabase method)
multi_get_tensor() (Data method)
(FeatureStore method)
(HeteroData method)
multi_insert() (Database method)
(RocksDatabase method)
(SQLiteDatabase method)
MultiAggregation (class in torch_geometric.nn.aggr)
MyketDataset (class in torch_geometric.datasets)
N
narrow() (in module torch_geometric.utils)
nearest() (in module torch_geometric.nn.pool)
neg_embedding_loss() (SignedGCN method)
negative_sampling() (in module torch_geometric.utils)
NegativeSampling (class in torch_geometric.sampler)
NeighborLoader (class in torch_geometric.loader)
NeighborSampler (class in torch_geometric.loader)
(class in torch_geometric.sampler)
NELL (class in torch_geometric.datasets)
NeuralFingerprint (class in torch_geometric.nn.models)
NeuroGraphDataset (class in torch_geometric.datasets)
nll_loss() (SignedGCN method)
NNConv (class in torch_geometric.nn.conv)
Node2Vec (class in torch_geometric.nn.models)
node_attrs() (Data method)
(HeteroData method)
(TemporalData method)
node_items() (HeteroData method)
node_sample() (DistNeighborSampler method)
node_stores (HeteroData property)
node_type_subgraph() (HeteroData method)
node_types (HeteroData property)
NodeLoader (class in torch_geometric.loader)
NodePropertySplit (class in torch_geometric.transforms)
NodeSamplerInput (class in torch_geometric.sampler)
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)
(OnDiskDataset property)
num_cols (EdgeIndex property)
num_edge_features (Data property)
(Dataset property)
(HeteroData property)
(InMemoryDataset property)
(OnDiskDataset property)
num_edge_types (Data 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)
(InMemoryDataset property)
(OnDiskDataset property)
num_graphs (Batch property)
num_hops (NumNeighbors property)
num_layers (MLP property)
num_node_features (Data property)
(Dataset property)
(HeteroData property)
(InMemoryDataset property)
(OnDiskDataset property)
num_node_types (Data property)
num_nodes (Data property)
(HeteroData property)
(TemporalData property)
num_rows (EdgeIndex property)
numel (KNNIndex property)
NumNeighbors (class in torch_geometric.sampler)
O
OGB_MAG (class in torch_geometric.datasets)
OMDB (class in torch_geometric.datasets)
OnDiskDataset (class in torch_geometric.data)
one_hot() (in module torch_geometric.utils)
OneHotDegree (class in torch_geometric.transforms)
OSE_GVCS (class in torch_geometric.datasets)
out_channels (MLP property)
output() (Transformer method)
P
Pad (class in torch_geometric.transforms)
PairNorm (class in torch_geometric.nn.norm)
PANConv (class in torch_geometric.nn.conv)
PANPooling (class in torch_geometric.nn.pool)
params_count() (in module torch_geometric.graphgym.utils)
parse_args() (in module torch_geometric.graphgym)
Partitioner (class in torch_geometric.distributed)
PascalPF (class in torch_geometric.datasets)
PascalVOCKeypoints (class in torch_geometric.datasets)
PCPNetDataset (class in torch_geometric.datasets)
PCQM4Mv2 (class in torch_geometric.datasets)
PDNConv (class in torch_geometric.nn.conv)
PGExplainer (class in torch_geometric.explain.algorithm)
PGMExplainer (class in torch_geometric.contrib.explain)
pin_memory() (Data method)
(HeteroData method)
(TemporalData method)
placeholder() (Transformer method)
Planetoid (class in torch_geometric.datasets)
PMLP (class in torch_geometric.nn.models)
PNA (class in torch_geometric.nn.models)
PNAConv (class in torch_geometric.nn.conv)
PointGNNConv (class in 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)
PositionalEncoding (class in torch_geometric.nn.encoding)
PowerMeanAggregation (class in torch_geometric.nn.aggr)
PPFConv (class in torch_geometric.nn.conv)
PPI (class in torch_geometric.datasets)
PRBCDAttack (class in torch_geometric.contrib.nn.models)
pre_transform() (RENet static method)
predict_link() (LightGCN method)
PrefetchLoader (class in torch_geometric.loader)
print_summary() (Dataset method)
(InMemoryDataset method)
(OnDiskDataset method)
print_time_total() (in module torch_geometric.profile)
process() (Dataset method)
(InMemoryDataset method)
(OnDiskDataset method)
processed_file_names (Dataset property)
(InMemoryDataset property)
(OnDiskDataset property)
processed_paths (Dataset property)
(InMemoryDataset property)
(OnDiskDataset property)
profileit() (in module torch_geometric.profile)
propagate() (MessagePassing method)
put_edge_index() (Data method)
(GraphStore method)
(HeteroData method)
put_tensor() (Data method)
(FeatureStore method)
(HeteroData method)
PyGModelHubMixin (class in torch_geometric.nn.model_hub)
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)
random_sample() (KGEModel method)
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)
(OnDiskDataset property)
raw_paths (Dataset property)
(InMemoryDataset property)
(OnDiskDataset property)
RCDD (class in torch_geometric.datasets)
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_edge_index() (Data method)
(GraphStore method)
(HeteroData method)
remove_isolated_nodes() (in module torch_geometric.utils)
remove_self_loops() (in module torch_geometric.utils)
remove_tensor() (Data method)
(FeatureStore method)
(HeteroData method)
RemoveDuplicatedEdges (class in torch_geometric.transforms)
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)
replace_all_uses_with() (Transformer method)
requires_grad_() (Data method)
(HeteroData method)
(TemporalData method)
reset() (LinkPredF1 method)
(LinkPredMAP method)
(LinkPredNDCG method)
(LinkPredPrecision method)
(LinkPredRecall method)
(timeit method)
reset_parameters() (Aggregation method)
(AGNNConv method)
(AntiSymmetricConv method)
(APPNP method)
(ARGA method)
(ARGVA method)
(ARMAConv method)
(ASAPooling method)
(AttentionalAggregation method)
(AttentiveFP method)
(BatchNorm method)
(CGConv method)
(ChebConv method)
(ClusterGCNConv method)
(ComplEx method)
(CuGraphGATConv method)
(CuGraphRGCNConv method)
(CuGraphSAGEConv method)
(DeepGCNLayer method)
(DeepGraphInfomax method)
(DeepSetsAggregation method)
(DegreeScalerAggregation method)
(DenseGCNConv method)
(DenseGINConv method)
(DenseGraphConv method)
(DenseSAGEConv method)
(DiffGroupNorm method)
(DimeNet method)
(DimeNetPlusPlus method)
(DirGNNConv method)
(DistMult method)
(DMoNPooling method)
(DNAConv method)
(DynamicEdgeConv method)
(EdgeCNN method)
(EdgeConv method)
(EdgePooling method)
(EGConv method)
(EquilibriumAggregation method)
(FAConv method)
(FastRGCNConv method)
(FeaStConv method)
(FiLMConv method)
(FusedGATConv method)
(GAE method)
(GAT method)
(GATConv method)
(GatedGraphConv method)
(GATv2Conv method)
(GCN method)
(GCN2Conv method)
(GCNConv method)
(GENConv method)
(GeneralConv method)
(GIN method)
(GINConv method)
(GINEConv method)
(GMMConv method)
(GNNFF method)
(GPSConv method)
(GraphConv method)
(GraphMaskExplainer method)
(GraphMultisetTransformer method)
(GraphNorm method)
(GraphSAGE method)
(GraphUNet method)
(GravNetConv method)
(GroupAddRev method)
(GRUAggregation method)
(HANConv method)
(HEATConv method)
(HeteroBatchNorm method)
(HeteroConv method)
(HeteroDictLinear method)
(HeteroLayerNorm method)
(HeteroLinear method)
(HGTConv method)
(HypergraphConv method)
(InstanceNorm method)
(JumpingKnowledge method)
(KGEModel method)
(LabelPropagation method)
(LayerNorm method)
(LCMAggregation method)
(LEConv method)
(LGConv method)
(LightGCN method)
(Linear method)
(LINKX method)
(LSTMAggregation method)
(MaskLabel method)
(MemPooling method)
(MessageNorm method)
(MessagePassing method)
(MetaLayer method)
(MetaPath2Vec method)
(MFConv method)
(MixHopConv method)
(MLP method)
(MLPAggregation method)
(MultiAggregation method)
(NeuralFingerprint method)
(NNConv method)
(Node2Vec method)
(PANConv method)
(PANPooling method)
(PDNConv method)
(PGExplainer method)
(PMLP method)
(PNA method)
(PNAConv method)
(PointGNNConv method)
(PointNetConv method)
(PointTransformerConv method)
(PositionalEncoding method)
(PowerMeanAggregation method)
(PPFConv method)
(RECT_L method)
(RENet method)
(ResGatedGraphConv method)
(RGATConv method)
(RGCNConv method)
(RotatE method)
(SAGEConv method)
(SAGPooling method)
(SchNet method)
(Set2Set method)
(SetTransformerAggregation method)
(SGConv method)
(SignedConv method)
(SignedGCN method)
(SimpleConv method)
(SoftmaxAggregation method)
(SplineConv method)
(SSGConv method)
(SuperGATConv method)
(TAGConv method)
(TemporalEncoding method)
(TGNMemory method)
(TopKPooling method)
(TransE method)
(TransformerConv method)
(VGAE method)
(ViSNet method)
(WLConv method)
(WLConvContinuous method)
(XConv method)
reset_running_stats() (BatchNorm method)
(HeteroBatchNorm 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)
RocksDatabase (class in torch_geometric.data)
RootedEgoNets (class in torch_geometric.transforms)
RootedRWSubgraph (class in torch_geometric.transforms)
RotatE (class in torch_geometric.nn.kge)
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() (NegativeSampling method)
sample_from_edges() (BaseSampler method)
sample_from_nodes() (BaseSampler method)
sample_one_hop() (DistNeighborSampler method)
SamplePoints (class in torch_geometric.transforms)
SamplerOutput (class in torch_geometric.sampler)
save() (InMemoryDataset class method)
save_ckpt() (in module torch_geometric.graphgym)
save_pretrained() (PyGModelHubMixin method)
scatter() (in module torch_geometric.utils)
SchNet (class in torch_geometric.nn.models)
search() (KNNIndex method)
seed_everything() (in module torch_geometric.seed)
segment() (in module torch_geometric.utils)
segregate_self_loops() (in module torch_geometric.utils)
select() (in module torch_geometric.utils)
Sequential (class in torch_geometric.nn.sequential)
serialize() (OnDiskDataset method)
Set2Set (class in torch_geometric.nn.aggr)
set_cfg() (in module torch_geometric.graphgym)
set_debug (class in torch_geometric.debug)
set_experimental_mode (class in torch_geometric.experimental)
set_home_dir() (in module torch_geometric.home)
set_out_dir() (in module torch_geometric.graphgym)
set_printing() (in module torch_geometric.graphgym)
set_run_dir() (in module torch_geometric.graphgym)
set_value_dict() (HeteroData method)
SetTransformerAggregation (class in torch_geometric.nn.aggr)
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)
(InMemoryDataset method)
(OnDiskDataset 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)
SimpleConv (class in torch_geometric.nn.conv)
size() (Data method)
(HeteroData method)
(TemporalData method)
smooth() (CorrectAndSmooth method)
SNAPDataset (class in torch_geometric.datasets)
snapshot() (Data method)
(HeteroData method)
(TemporalData method)
softmax() (in module torch_geometric.utils)
SoftmaxAggregation (class in torch_geometric.nn.aggr)
sort() (Data method)
(HeteroData method)
(TemporalData method)
sort_by() (EdgeIndex method)
sort_by_time() (Data method)
(HeteroData method)
(TemporalData method)
sort_edge_index() (in module torch_geometric.utils)
sort_order (EdgeIndex property)
SortAggregation (class in torch_geometric.nn.aggr)
sparse_narrow() (EdgeIndex method)
sparse_resize_() (EdgeIndex method)
sparse_size() (EdgeIndex 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)
SQLiteDatabase (class in torch_geometric.data)
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)
summary() (in module torch_geometric.nn.summary)
SuperGATConv (class in torch_geometric.nn.conv)
supports() (AttentionExplainer method)
(CaptumExplainer method)
(DummyExplainer method)
(ExplainerAlgorithm method)
(GNNExplainer method)
(GraphMaskExplainer method)
(PGExplainer method)
(PGMExplainer method)
SVDFeatureReduction (class in torch_geometric.transforms)
T
TAGConv (class in torch_geometric.nn.conv)
Taobao (class in torch_geometric.datasets)
TargetIndegree (class in torch_geometric.transforms)
TemporalData (class in torch_geometric.data)
TemporalDataLoader (class in torch_geometric.loader)
TemporalEncoding (class in torch_geometric.nn.encoding)
TensorAttr (class in torch_geometric.data)
test() (DeepGraphInfomax method)
(GAE method)
(KGEModel 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)
(InMemoryDataset method)
(TemporalData method)
to_bidirectional() (HeteroSamplerOutput method)
(SamplerOutput method)
to_captum_input (class in torch_geometric.nn.models)
to_captum_model (class in torch_geometric.nn.models)
to_cugraph() (in module torch_geometric.utils)
to_data_list() (Batch method)
to_datapipe() (Dataset method)
(InMemoryDataset method)
(OnDiskDataset method)
to_dense() (EdgeIndex method)
to_dense_adj() (in module torch_geometric.utils)
to_dense_batch() (in module torch_geometric.utils)
to_dgl() (in module torch_geometric.utils)
to_dict() (Data method)
(HeteroData method)
(TemporalData method)
to_edge_index() (in module torch_geometric.utils)
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_nested_tensor() (in module torch_geometric.utils)
to_networkit() (in module torch_geometric.utils)
to_networkx() (in module torch_geometric.utils)
to_on_disk_dataset() (InMemoryDataset method)
to_scipy_sparse_matrix() (in module torch_geometric.utils)
to_smiles() (in module torch_geometric.utils)
to_sparse() (EdgeIndex method)
to_sparse_coo() (EdgeIndex method)
to_sparse_csc() (EdgeIndex method)
to_sparse_csr() (EdgeIndex method)
to_sparse_tensor() (EdgeIndex method)
to_torch_coo_tensor() (in module torch_geometric.utils)
to_torch_csc_tensor() (in module torch_geometric.utils)
to_torch_csr_tensor() (in module torch_geometric.utils)
to_torch_sparse_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.compile
module
torch_geometric.contrib.datasets
module
torch_geometric.contrib.explain
module
torch_geometric.contrib.nn.conv
module
torch_geometric.contrib.nn.models
module
torch_geometric.contrib.transforms
module
torch_geometric.debug
module
torch_geometric.distributed
module
torch_geometric.experimental
module
torch_geometric.graphgym
module
torch_geometric.graphgym.models
module
torch_geometric.graphgym.utils
module
torch_geometric.home
module
torch_geometric.loader
module
torch_geometric.nn.data_parallel
module
torch_geometric.nn.encoding
module
torch_geometric.nn.model_hub
module
torch_geometric.nn.summary
module
torch_geometric.profile
module
torch_geometric.sampler
module
torch_geometric.seed
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)
train() (in module torch_geometric.graphgym)
(PGExplainer method)
(TGNMemory method)
train_test_split_edges() (in module torch_geometric.utils)
train_val_test_split() (TemporalData method)
TransE (class in torch_geometric.nn.kge)
transform() (Transformer method)
Transformer (class in torch_geometric.nn.fx)
TransformerConv (class in torch_geometric.nn.conv)
tree_decomposition() (in module torch_geometric.utils)
TreeGraph (class in torch_geometric.datasets.graph_generator)
trim_to_layer() (in module torch_geometric.utils)
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)
unfaithfulness() (in module torch_geometric.explain.metric)
unpool() (EdgePooling method)
up_to() (Data method)
(HeteroData method)
(TemporalData method)
update() (Data method)
(HeteroData method)
(LinkPredF1 method)
(LinkPredMAP method)
(LinkPredNDCG method)
(LinkPredPrecision method)
(LinkPredRecall method)
(MessagePassing method)
(TemporalData method)
(TensorAttr method)
update_state() (TGNMemory method)
update_tensor() (Data method)
(FeatureStore method)
(HeteroData method)
UPFD (class in torch_geometric.datasets)
V
validate() (Data method)
(EdgeIndex method)
(Explanation method)
(HeteroData method)
(HeteroExplanation method)
VarAggregation (class in torch_geometric.nn.aggr)
VariancePreservingAggregation (class in torch_geometric.nn.aggr)
VGAE (class in torch_geometric.nn.models)
view() (Data method)
(FeatureStore method)
(HeteroData method)
VirtualNode (class in torch_geometric.transforms)
ViSNet (class in torch_geometric.nn.models)
visualize_feature_importance() (Explanation method)
(HeteroExplanation method)
visualize_graph() (Explanation method)
voxel_grid() (in module torch_geometric.nn.pool)
W
WebKB (class in torch_geometric.datasets)
WikiCS (class in torch_geometric.datasets)
Wikidata5M (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)
xpu_profile() (in module torch_geometric.profile)
Y
Yelp (class in torch_geometric.datasets)
Z
ZINC (class in torch_geometric.datasets)
ZipLoader (class in torch_geometric.loader)
Read the Docs
v: latest
Versions
latest
stable
2.5.2
2.5.1
2.5.0
2.4.0
2.3.1
2.3.0
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