All modules for which code is available
- torch_geometric.data.batch
- torch_geometric.data.data
- torch_geometric.data.dataset
- torch_geometric.data.download
- torch_geometric.data.extract
- torch_geometric.data.hetero_data
- torch_geometric.data.in_memory_dataset
- torch_geometric.data.lightning_datamodule
- torch_geometric.data.makedirs
- torch_geometric.data.temporal
- torch_geometric.datasets.actor
- torch_geometric.datasets.airports
- torch_geometric.datasets.amazon
- torch_geometric.datasets.amazon_products
- torch_geometric.datasets.aminer
- torch_geometric.datasets.aqsol
- torch_geometric.datasets.attributed_graph_dataset
- torch_geometric.datasets.ba_shapes
- torch_geometric.datasets.bitcoin_otc
- torch_geometric.datasets.citation_full
- torch_geometric.datasets.coauthor
- torch_geometric.datasets.coma
- torch_geometric.datasets.dblp
- torch_geometric.datasets.dbp15k
- torch_geometric.datasets.deezer_europe
- torch_geometric.datasets.dgraph
- torch_geometric.datasets.dynamic_faust
- torch_geometric.datasets.elliptic
- torch_geometric.datasets.email_eu_core
- torch_geometric.datasets.entities
- torch_geometric.datasets.facebook
- torch_geometric.datasets.fake
- torch_geometric.datasets.faust
- torch_geometric.datasets.flickr
- torch_geometric.datasets.gdelt
- torch_geometric.datasets.ged_dataset
- torch_geometric.datasets.gemsec
- torch_geometric.datasets.geometry
- torch_geometric.datasets.github
- torch_geometric.datasets.gnn_benchmark_dataset
- torch_geometric.datasets.hgb_dataset
- torch_geometric.datasets.hydro_net
- torch_geometric.datasets.icews
- torch_geometric.datasets.imdb
- torch_geometric.datasets.jodie
- torch_geometric.datasets.karate
- torch_geometric.datasets.last_fm
- torch_geometric.datasets.lastfm_asia
- torch_geometric.datasets.linkx_dataset
- torch_geometric.datasets.lrgb
- torch_geometric.datasets.malnet_tiny
- torch_geometric.datasets.md17
- torch_geometric.datasets.mixhop_synthetic_dataset
- torch_geometric.datasets.mnist_superpixels
- torch_geometric.datasets.modelnet
- torch_geometric.datasets.molecule_net
- torch_geometric.datasets.movie_lens
- torch_geometric.datasets.nell
- torch_geometric.datasets.ogb_mag
- torch_geometric.datasets.omdb
- torch_geometric.datasets.pascal
- torch_geometric.datasets.pascal_pf
- torch_geometric.datasets.pcpnet_dataset
- torch_geometric.datasets.planetoid
- torch_geometric.datasets.polblogs
- torch_geometric.datasets.ppi
- torch_geometric.datasets.qm7
- torch_geometric.datasets.qm9
- torch_geometric.datasets.reddit
- torch_geometric.datasets.reddit2
- torch_geometric.datasets.rel_link_pred_dataset
- torch_geometric.datasets.s3dis
- torch_geometric.datasets.sbm_dataset
- torch_geometric.datasets.shapenet
- torch_geometric.datasets.shrec2016
- torch_geometric.datasets.snap_dataset
- torch_geometric.datasets.suite_sparse
- torch_geometric.datasets.tosca
- torch_geometric.datasets.tu_dataset
- torch_geometric.datasets.twitch
- torch_geometric.datasets.upfd
- torch_geometric.datasets.webkb
- torch_geometric.datasets.wikics
- torch_geometric.datasets.wikipedia_network
- torch_geometric.datasets.willow_object_class
- torch_geometric.datasets.word_net
- torch_geometric.datasets.yelp
- torch_geometric.datasets.zinc
- torch_geometric.debug
- torch_geometric.explain.algorithm.base
- torch_geometric.explain.algorithm.dummy_explainer
- torch_geometric.explain.algorithm.gnn_explainer
- torch_geometric.explain.config
- torch_geometric.explain.explainer
- torch_geometric.explain.explanation
- torch_geometric.graphgym.checkpoint
- torch_geometric.graphgym.cmd_args
- torch_geometric.graphgym.config
- torch_geometric.graphgym.init
- torch_geometric.graphgym.loader
- torch_geometric.graphgym.logger
- torch_geometric.graphgym.loss
- torch_geometric.graphgym.model_builder
- torch_geometric.graphgym.models.encoder
- torch_geometric.graphgym.models.gnn
- torch_geometric.graphgym.models.head
- torch_geometric.graphgym.models.layer
- torch_geometric.graphgym.optim
- torch_geometric.graphgym.register
- torch_geometric.graphgym.utils.agg_runs
- torch_geometric.graphgym.utils.comp_budget
- torch_geometric.graphgym.utils.device
- torch_geometric.graphgym.utils.epoch
- torch_geometric.graphgym.utils.io
- torch_geometric.graphgym.utils.tools
- torch_geometric.loader.cluster
- torch_geometric.loader.data_list_loader
- torch_geometric.loader.dataloader
- torch_geometric.loader.dense_data_loader
- torch_geometric.loader.dynamic_batch_sampler
- torch_geometric.loader.graph_saint
- torch_geometric.loader.hgt_loader
- torch_geometric.loader.imbalanced_sampler
- torch_geometric.loader.link_loader
- torch_geometric.loader.link_neighbor_loader
- torch_geometric.loader.neighbor_loader
- torch_geometric.loader.neighbor_sampler
- torch_geometric.loader.node_loader
- torch_geometric.loader.random_node_loader
- torch_geometric.loader.shadow
- torch_geometric.loader.temporal_dataloader
- torch_geometric.nn.aggr.attention
- torch_geometric.nn.aggr.base
- torch_geometric.nn.aggr.basic
- torch_geometric.nn.aggr.equilibrium
- torch_geometric.nn.aggr.gmt
- torch_geometric.nn.aggr.lstm
- torch_geometric.nn.aggr.multi
- torch_geometric.nn.aggr.quantile
- torch_geometric.nn.aggr.scaler
- torch_geometric.nn.aggr.set2set
- torch_geometric.nn.aggr.sort
- torch_geometric.nn.conv.agnn_conv
- torch_geometric.nn.conv.appnp
- torch_geometric.nn.conv.arma_conv
- torch_geometric.nn.conv.cg_conv
- torch_geometric.nn.conv.cheb_conv
- torch_geometric.nn.conv.cluster_gcn_conv
- torch_geometric.nn.conv.dna_conv
- torch_geometric.nn.conv.edge_conv
- torch_geometric.nn.conv.eg_conv
- torch_geometric.nn.conv.fa_conv
- torch_geometric.nn.conv.feast_conv
- torch_geometric.nn.conv.film_conv
- torch_geometric.nn.conv.fused_gat_conv
- torch_geometric.nn.conv.gat_conv
- torch_geometric.nn.conv.gated_graph_conv
- torch_geometric.nn.conv.gatv2_conv
- torch_geometric.nn.conv.gcn2_conv
- torch_geometric.nn.conv.gcn_conv
- torch_geometric.nn.conv.gen_conv
- torch_geometric.nn.conv.general_conv
- torch_geometric.nn.conv.gin_conv
- torch_geometric.nn.conv.gmm_conv
- torch_geometric.nn.conv.graph_conv
- torch_geometric.nn.conv.gravnet_conv
- torch_geometric.nn.conv.han_conv
- torch_geometric.nn.conv.heat_conv
- torch_geometric.nn.conv.hetero_conv
- torch_geometric.nn.conv.hgt_conv
- torch_geometric.nn.conv.hypergraph_conv
- torch_geometric.nn.conv.le_conv
- torch_geometric.nn.conv.lg_conv
- torch_geometric.nn.conv.message_passing
- torch_geometric.nn.conv.mf_conv
- torch_geometric.nn.conv.nn_conv
- torch_geometric.nn.conv.pan_conv
- torch_geometric.nn.conv.pdn_conv
- torch_geometric.nn.conv.pna_conv
- torch_geometric.nn.conv.point_conv
- torch_geometric.nn.conv.point_transformer_conv
- torch_geometric.nn.conv.ppf_conv
- torch_geometric.nn.conv.res_gated_graph_conv
- torch_geometric.nn.conv.rgat_conv
- torch_geometric.nn.conv.rgcn_conv
- torch_geometric.nn.conv.sage_conv
- torch_geometric.nn.conv.sg_conv
- torch_geometric.nn.conv.signed_conv
- torch_geometric.nn.conv.spline_conv
- torch_geometric.nn.conv.ssg_conv
- torch_geometric.nn.conv.supergat_conv
- torch_geometric.nn.conv.tag_conv
- torch_geometric.nn.conv.transformer_conv
- torch_geometric.nn.conv.wl_conv
- torch_geometric.nn.conv.wl_conv_continuous
- torch_geometric.nn.conv.x_conv
- torch_geometric.nn.data_parallel
- torch_geometric.nn.dense.dense_gcn_conv
- torch_geometric.nn.dense.dense_gin_conv
- torch_geometric.nn.dense.dense_graph_conv
- torch_geometric.nn.dense.dense_sage_conv
- torch_geometric.nn.dense.diff_pool
- torch_geometric.nn.dense.dmon_pool
- torch_geometric.nn.dense.linear
- torch_geometric.nn.dense.mincut_pool
- torch_geometric.nn.encoding
- torch_geometric.nn.functional.bro
- torch_geometric.nn.functional.gini
- torch_geometric.nn.fx
- torch_geometric.nn.meta
- torch_geometric.nn.models.attentive_fp
- torch_geometric.nn.models.autoencoder
- torch_geometric.nn.models.basic_gnn
- torch_geometric.nn.models.captum
- torch_geometric.nn.models.correct_and_smooth
- torch_geometric.nn.models.deep_graph_infomax
- torch_geometric.nn.models.deepgcn
- torch_geometric.nn.models.dimenet
- torch_geometric.nn.models.graph_unet
- torch_geometric.nn.models.jumping_knowledge
- torch_geometric.nn.models.label_prop
- torch_geometric.nn.models.lightgcn
- torch_geometric.nn.models.linkx
- torch_geometric.nn.models.mask_label
- torch_geometric.nn.models.metapath2vec
- torch_geometric.nn.models.mlp
- torch_geometric.nn.models.node2vec
- torch_geometric.nn.models.re_net
- torch_geometric.nn.models.rect
- torch_geometric.nn.models.rev_gnn
- torch_geometric.nn.models.schnet
- torch_geometric.nn.models.signed_gcn
- torch_geometric.nn.models.tgn
- torch_geometric.nn.norm.batch_norm
- torch_geometric.nn.norm.diff_group_norm
- torch_geometric.nn.norm.graph_norm
- torch_geometric.nn.norm.graph_size_norm
- torch_geometric.nn.norm.instance_norm
- torch_geometric.nn.norm.layer_norm
- torch_geometric.nn.norm.mean_subtraction_norm
- torch_geometric.nn.norm.msg_norm
- torch_geometric.nn.norm.pair_norm
- torch_geometric.nn.pool
- torch_geometric.nn.pool.asap
- torch_geometric.nn.pool.avg_pool
- torch_geometric.nn.pool.edge_pool
- torch_geometric.nn.pool.glob
- torch_geometric.nn.pool.graclus
- torch_geometric.nn.pool.max_pool
- torch_geometric.nn.pool.mem_pool
- torch_geometric.nn.pool.pan_pool
- torch_geometric.nn.pool.sag_pool
- torch_geometric.nn.pool.topk_pool
- torch_geometric.nn.pool.voxel_grid
- torch_geometric.nn.sequential
- torch_geometric.nn.to_hetero_transformer
- torch_geometric.nn.to_hetero_with_bases_transformer
- torch_geometric.nn.unpool.knn_interpolate
- torch_geometric.profile.profile
- torch_geometric.profile.utils
- torch_geometric.sampler.base
- torch_geometric.sampler.hgt_sampler
- torch_geometric.sampler.neighbor_sampler
- torch_geometric.seed
- torch_geometric.transforms.add_metapaths
- torch_geometric.transforms.add_positional_encoding
- torch_geometric.transforms.add_self_loops
- torch_geometric.transforms.base_transform
- torch_geometric.transforms.cartesian
- torch_geometric.transforms.center
- torch_geometric.transforms.compose
- torch_geometric.transforms.constant
- torch_geometric.transforms.delaunay
- torch_geometric.transforms.distance
- torch_geometric.transforms.face_to_edge
- torch_geometric.transforms.feature_propagation
- torch_geometric.transforms.fixed_points
- torch_geometric.transforms.gcn_norm
- torch_geometric.transforms.gdc
- torch_geometric.transforms.generate_mesh_normals
- torch_geometric.transforms.grid_sampling
- torch_geometric.transforms.knn_graph
- torch_geometric.transforms.laplacian_lambda_max
- torch_geometric.transforms.largest_connected_components
- torch_geometric.transforms.line_graph
- torch_geometric.transforms.linear_transformation
- torch_geometric.transforms.local_cartesian
- torch_geometric.transforms.local_degree_profile
- torch_geometric.transforms.mask
- torch_geometric.transforms.normalize_features
- torch_geometric.transforms.normalize_rotation
- torch_geometric.transforms.normalize_scale
- torch_geometric.transforms.one_hot_degree
- torch_geometric.transforms.point_pair_features
- torch_geometric.transforms.polar
- torch_geometric.transforms.radius_graph
- torch_geometric.transforms.random_flip
- torch_geometric.transforms.random_jitter
- torch_geometric.transforms.random_link_split
- torch_geometric.transforms.random_node_split
- torch_geometric.transforms.random_rotate
- torch_geometric.transforms.random_scale
- torch_geometric.transforms.random_shear
- torch_geometric.transforms.remove_isolated_nodes
- torch_geometric.transforms.remove_training_classes
- torch_geometric.transforms.rooted_subgraph
- torch_geometric.transforms.sample_points
- torch_geometric.transforms.sign
- torch_geometric.transforms.spherical
- torch_geometric.transforms.svd_feature_reduction
- torch_geometric.transforms.target_indegree
- torch_geometric.transforms.to_dense
- torch_geometric.transforms.to_device
- torch_geometric.transforms.to_sparse_tensor
- torch_geometric.transforms.to_superpixels
- torch_geometric.transforms.to_undirected
- torch_geometric.transforms.two_hop
- torch_geometric.transforms.virtual_node
- torch_geometric.utils.assortativity
- torch_geometric.utils.augmentation
- torch_geometric.utils.coalesce
- torch_geometric.utils.convert
- torch_geometric.utils.degree
- torch_geometric.utils.dropout
- torch_geometric.utils.geodesic
- torch_geometric.utils.get_laplacian
- torch_geometric.utils.get_mesh_laplacian
- torch_geometric.utils.grid
- torch_geometric.utils.homophily
- torch_geometric.utils.isolated
- torch_geometric.utils.loop
- torch_geometric.utils.mask
- torch_geometric.utils.negative_sampling
- torch_geometric.utils.normalized_cut
- torch_geometric.utils.random
- torch_geometric.utils.smiles
- torch_geometric.utils.softmax
- torch_geometric.utils.sort_edge_index
- torch_geometric.utils.sparse
- torch_geometric.utils.spmm
- torch_geometric.utils.subgraph
- torch_geometric.utils.to_dense_adj
- torch_geometric.utils.to_dense_batch
- torch_geometric.utils.train_test_split_edges
- torch_geometric.utils.tree_decomposition
- torch_geometric.utils.unbatch
- torch_geometric.utils.undirected