JittorGeometric Logo

Getting Started

  • Installation
    • System Requirements
      • GPU Environment (optional)
      • Ascend Environment (optional)
    • Dependencies
    • Installation Steps
    • Troubleshooting
  • Get Started
    • Quick Start
    • Step 1: Import Libraries
    • Step 2: Load a Dataset
    • Step 3: Define a Simple GCN Model
    • Step 4: Training the Model
    • Step 5: Evaluate the Model
    • Next Steps

Core Modules

  • Neural Network Modules
    • jittor_geometric.nn.aggr
      • Aggregation
        • Aggregation.__init__()
        • Aggregation.execute()
        • Aggregation.reset_parameters()
        • Aggregation.assert_index_present()
        • Aggregation.assert_sorted_index()
        • Aggregation.assert_two_dimensional_input()
        • Aggregation.reduce()
        • Aggregation.segment()
        • Aggregation.to_dense_batch()
        • Aggregation.to_dense_batch_helper()
      • MultiAggregation
        • MultiAggregation.__init__()
        • MultiAggregation.reset_parameters()
        • MultiAggregation.get_out_channels()
        • MultiAggregation.execute()
        • MultiAggregation.combine()
      • SumAggregation
        • SumAggregation.execute()
      • MeanAggregation
        • MeanAggregation.execute()
      • MaxAggregation
        • MaxAggregation.execute()
      • MinAggregation
        • MinAggregation.execute()
    • jittor_geometric.nn.conv
      • MessagePassing
        • MessagePassing.special_args
        • MessagePassing.__init__()
        • MessagePassing.propagate()
        • MessagePassing.message()
        • MessagePassing.aggregate()
        • MessagePassing.update()
      • GCNConv
        • GCNConv.__init__()
        • GCNConv.reset_parameters()
        • GCNConv.execute()
        • GCNConv.propagate_msg()
        • GCNConv.propagate_spmm()
      • ChebConv
        • ChebConv.__init__()
        • ChebConv.reset_parameters()
        • ChebConv.execute()
        • ChebConv.forward()
        • ChebConv.message()
      • SGConv
        • SGConv.__init__()
        • SGConv.reset_parameters()
        • SGConv.execute()
        • SGConv.propagate_msg()
        • SGConv.propagate_spmm()
      • GCN2Conv
        • GCN2Conv.__init__()
        • GCN2Conv.reset_parameters()
        • GCN2Conv.execute()
        • GCN2Conv.propagate_msg()
        • GCN2Conv.propagate_spmm()
      • MessagePassingNts
        • MessagePassingNts.special_args
        • MessagePassingNts.__init__()
        • MessagePassingNts.propagate()
        • MessagePassingNts.aggregate_with_weight()
        • MessagePassingNts.scatter_to_edge()
        • MessagePassingNts.edge_forward()
        • MessagePassingNts.scatter_to_vertex()
        • MessagePassingNts.vertex_forward()
      • GATConv
        • GATConv.__init__()
        • GATConv.reset_parameters()
        • GATConv.execute()
        • GATConv.propagate()
        • GATConv.scatter_to_edge()
        • GATConv.edge_forward()
        • GATConv.scatter_to_vertex()
        • GATConv.vertex_forward()
      • EGNNConv
        • EGNNConv.__init__()
        • EGNNConv.init_weights()
        • EGNNConv.execute()
        • EGNNConv.message()
        • EGNNConv.propagate()
      • APPNP
        • APPNP.__init__()
        • APPNP.reset_parameters()
        • APPNP.execute()
        • APPNP.propagate_msg()
        • APPNP.propagate_spmm()
      • GPRGNN
        • GPRGNN.__init__()
        • GPRGNN.reset_parameters()
        • GPRGNN.execute()
        • GPRGNN.propagate_msg()
        • GPRGNN.propagate_spmm()
      • EvenNet
        • EvenNet.__init__()
        • EvenNet.reset_parameters()
        • EvenNet.execute()
        • EvenNet.propagate_msg()
        • EvenNet.propagate_spmm()
      • BernNet
        • BernNet.__init__()
        • BernNet.reset_parameters()
        • BernNet.execute()
        • BernNet.propagate_msg()
        • BernNet.propagate_spmm()
      • ChebNetII
        • ChebNetII.__init__()
        • ChebNetII.reset_parameters()
        • ChebNetII.execute()
        • ChebNetII.propagate_msg()
        • ChebNetII.propagate_spmm()
      • TransformerConv
        • TransformerConv.__init__()
        • TransformerConv.execute()
        • TransformerConv.forward()
        • TransformerConv.message()
        • TransformerConv.reset_parameters()
      • OptBasisConv
        • OptBasisConv.__init__()
        • OptBasisConv.reset_parameters()
        • OptBasisConv.three_term_prop()
        • OptBasisConv.execute()
        • OptBasisConv.propagate_msg()
        • OptBasisConv.propagate_spmm()
      • ClusterGCNConv
        • ClusterGCNConv.__init__()
        • ClusterGCNConv.reset_parameters()
        • ClusterGCNConv.execute()
        • ClusterGCNConv.propagate_msg()
        • ClusterGCNConv.propagate_spmm()
      • SAGEConv
        • SAGEConv.__init__()
        • SAGEConv.reset_parameters()
        • SAGEConv.execute()
        • SAGEConv.propagate_msg()
        • SAGEConv.propagate_spmm()
    • jittor_geometric.nn.dense
      • MergeLayer
        • MergeLayer.__init__()
        • MergeLayer.execute()
      • TimeEncoder
        • TimeEncoder.__init__()
        • TimeEncoder.execute()
      • MLP
        • MLP.__init__()
        • MLP.execute()
  • jittor_geometric.data
    • Data
      • Data.__init__()
      • Data.from_dict()
      • Data.to_dict()
      • Data.to_namedtuple()
      • Data.__getitem__()
      • Data.__setitem__()
      • Data.keys
      • Data.__len__()
      • Data.__contains__()
      • Data.__iter__()
      • Data.__call__()
      • Data.__cat_dim__()
      • Data.__inc__()
      • Data.num_nodes
      • Data.num_edges
      • Data.num_faces
      • Data.num_node_features
      • Data.num_features
      • Data.num_edge_features
      • Data.contains_isolated_nodes()
      • Data.contains_self_loops()
      • Data.apply()
      • Data.contiguous()
      • Data.cpu()
      • Data.cuda()
      • Data.clone()
      • Data.pin_memory()
      • Data.debug()
    • Dataset
      • Dataset.raw_file_names
      • Dataset.processed_file_names
      • Dataset.download()
      • Dataset.process()
      • Dataset.len()
      • Dataset.get()
      • Dataset.__init__()
      • Dataset.indices()
      • Dataset.raw_dir
      • Dataset.processed_dir
      • Dataset.num_node_features
      • Dataset.num_features
      • Dataset.num_edge_features
      • Dataset.raw_paths
      • Dataset.processed_paths
      • Dataset.__len__()
      • Dataset.__getitem__()
      • Dataset.index_select()
      • Dataset.shuffle()
    • InMemoryDataset
      • InMemoryDataset.raw_file_names
      • InMemoryDataset.processed_file_names
      • InMemoryDataset.download()
      • InMemoryDataset.process()
      • InMemoryDataset.__init__()
      • InMemoryDataset.num_classes
      • InMemoryDataset.len()
      • InMemoryDataset.get()
      • InMemoryDataset.collate()
      • InMemoryDataset.copy()
    • download_url()
    • decide_download()
    • extract_zip()
    • extract_gz()
    • extract_tar()
    • CSC
      • CSC.__init__()
    • CSR
      • CSR.__init__()
    • GraphChunk
      • GraphChunk.__init__()
      • GraphChunk.set_csr()
      • GraphChunk.save()
      • GraphChunk.load()
    • TemporalData
      • TemporalData.__init__()
      • TemporalData.__setitem__()
      • TemporalData.keys
      • TemporalData.num_nodes
      • TemporalData.max_node_id
      • TemporalData.src_size
      • TemporalData.dst_size
      • TemporalData.num_events
      • TemporalData.num_edges
      • TemporalData.apply()
      • TemporalData.to()
      • TemporalData.train_val_test_split()
      • TemporalData.train_val_test_split_w_mask()
      • TemporalData.seq_batches()
    • Batch
      • Batch.from_data_list()
      • Batch.get_example()
      • Batch.to_data_list()
      • Batch.num_graphs
      • Batch.batch_size
    • Dictionary
      • Dictionary.__init__()
      • Dictionary.__len__()
      • Dictionary.vec_index()
      • Dictionary.index()
      • Dictionary.special_index()
      • Dictionary.add_symbol()
      • Dictionary.bos()
      • Dictionary.pad()
      • Dictionary.eos()
      • Dictionary.unk()
      • Dictionary.load()
      • Dictionary.add_from_file()
    • ConformerGen
      • ConformerGen.__init__()
      • ConformerGen.transform_raw()
  • jittor_geometric.datasets
    • Planetoid
      • Planetoid.url
      • Planetoid.__init__()
      • Planetoid.raw_dir
      • Planetoid.processed_dir
      • Planetoid.raw_file_names
      • Planetoid.processed_file_names
      • Planetoid.download()
      • Planetoid.process()
      • Planetoid.make_link_split()
    • Amazon
      • Amazon.url
      • Amazon.__init__()
      • Amazon.raw_dir
      • Amazon.processed_dir
      • Amazon.raw_file_names
      • Amazon.processed_file_names
      • Amazon.download()
      • Amazon.process()
      • Amazon.make_link_split()
    • WikipediaNetwork
      • WikipediaNetwork.url
      • WikipediaNetwork.__init__()
      • WikipediaNetwork.raw_dir
      • WikipediaNetwork.processed_dir
      • WikipediaNetwork.raw_file_names
      • WikipediaNetwork.processed_file_names
      • WikipediaNetwork.download()
      • WikipediaNetwork.process()
    • GeomGCN
      • GeomGCN.url
      • GeomGCN.__init__()
      • GeomGCN.raw_dir
      • GeomGCN.processed_dir
      • GeomGCN.raw_file_names
      • GeomGCN.processed_file_names
      • GeomGCN.download()
      • GeomGCN.process()
    • LINKXDataset
      • LINKXDataset.__init__()
      • LINKXDataset.raw_dir
      • LINKXDataset.processed_dir
      • LINKXDataset.raw_file_names
      • LINKXDataset.processed_file_names
      • LINKXDataset.download()
      • LINKXDataset.process()
    • OGBNodePropPredDataset
      • OGBNodePropPredDataset.__init__()
      • OGBNodePropPredDataset.get_idx_split()
      • OGBNodePropPredDataset.num_classes
      • OGBNodePropPredDataset.raw_file_names
      • OGBNodePropPredDataset.processed_file_names
      • OGBNodePropPredDataset.download()
      • OGBNodePropPredDataset.process()
    • HeteroDataset
      • HeteroDataset.url
      • HeteroDataset.__init__()
      • HeteroDataset.raw_dir
      • HeteroDataset.processed_dir
      • HeteroDataset.raw_file_names
      • HeteroDataset.processed_file_names
      • HeteroDataset.download()
      • HeteroDataset.process()
    • JODIEDataset
      • JODIEDataset.url
      • JODIEDataset.names
      • JODIEDataset.__init__()
      • JODIEDataset.raw_dir
      • JODIEDataset.processed_dir
      • JODIEDataset.raw_file_names
      • JODIEDataset.processed_file_names
      • JODIEDataset.download()
      • JODIEDataset.process()
    • Reddit
      • Reddit.url
      • Reddit.__init__()
      • Reddit.raw_file_names
      • Reddit.processed_file_names
      • Reddit.download()
      • Reddit.process()
    • TemporalDataLoader
      • TemporalDataLoader.__init__()
    • QM9
      • QM9.raw_url
      • QM9.raw_url2
      • QM9.__init__()
      • QM9.mean()
      • QM9.std()
      • QM9.atomref()
      • QM9.raw_file_names
      • QM9.processed_file_names
      • QM9.download()
      • QM9.process()
      • QM9.get_idx_split()
    • MoleculeNet
      • MoleculeNet.url
      • MoleculeNet.names
      • MoleculeNet.__init__()
      • MoleculeNet.raw_dir
      • MoleculeNet.processed_dir
      • MoleculeNet.raw_file_names
      • MoleculeNet.processed_file_names
      • MoleculeNet.download()
      • MoleculeNet.process()
    • MD17
      • MD17.gdml_url
      • MD17.revised_url
      • MD17.file_names
      • MD17.__init__()
      • MD17.mean()
      • MD17.raw_dir
      • MD17.processed_dir
      • MD17.raw_file_names
      • MD17.processed_file_names
      • MD17.download()
      • MD17.process()
    • PCQM4Mv2
      • PCQM4Mv2.url
      • PCQM4Mv2.split_mapping
      • PCQM4Mv2.__init__()
      • PCQM4Mv2.raw_file_names
      • PCQM4Mv2.processed_file_names
      • PCQM4Mv2.download()
      • PCQM4Mv2.process()
    • MovieLens1M
      • MovieLens1M.url
      • MovieLens1M.__init__()
      • MovieLens1M.raw_file_names
      • MovieLens1M.download()
      • MovieLens1M.read_raw()
    • MovieLens100K
      • MovieLens100K.url
      • MovieLens100K.__init__()
      • MovieLens100K.raw_file_names
      • MovieLens100K.download()
      • MovieLens100K.read_raw()
    • Yelp2018
      • Yelp2018.url
      • Yelp2018.raw_file_names
      • Yelp2018.__init__()
      • Yelp2018.download()
      • Yelp2018.read_raw()
  • jittor_geometric.dataloader
    • GeneralLoader
      • GeneralLoader.itermax
      • GeneralLoader.__init__()
      • GeneralLoader.itercnt
      • GeneralLoader.dataset
    • RandomNodeLoader
      • RandomNodeLoader.__init__()
      • RandomNodeLoader.get_node_indices()
    • NeighborLoader
      • NeighborLoader.__init__()
      • NeighborLoader.get_node_indices()
      • NeighborLoader.neighbor_sampling()
      • NeighborLoader.unique()
    • ClusterLoader
      • ClusterLoader.__init__()
      • ClusterLoader.partition()
      • ClusterLoader.get_node_indices()
    • DataLoader
      • DataLoader.__init__()

Advanced Features

  • jittor_geometric.partition
    • ChunkManager
      • ChunkManager.__init__()
      • ChunkManager.metis_partition()
      • ChunkManager.partition_to_chunk()
  • jittor_geometric.transforms
    • NormalizeFeatures
  • jittor_geometric.ops
  • 分布式训练
    • 第 0 步：设置分布式文件系统
    • 第 1 步：设置 IP 配置文件
    • 第 2 步：对图进行分区
    • 第 3 步：启动分布式作业
    • 单机多卡训练

Utilities & I/O

  • jittor_geometric.io
    • parse_txt_array()
    • read_txt_array()
    • read_planetoid_data()
    • read_npz()
    • read_graph()
    • read_heterograph()
    • read_node_label_hetero()
    • read_nodesplitidx_split_hetero()
  • jittor_geometric.utils
    • coalesce()
    • degree()
    • contains_self_loops()
    • remove_self_loops()
    • segregate_self_loops()
    • add_self_loops()
    • add_remaining_self_loops()
    • contains_isolated_nodes()
    • remove_isolated_nodes()
    • get_laplacian()
    • is_jittor_sparse_tensor()
    • scatter()
    • induced_graph()
    • unique()
    • neighbor_sampler()
    • randomwalk_sampler()
    • one_hot()
    • maybe_num_nodes()
    • from_rdmol()
    • to_rdmol()
    • from_smiles()
    • to_smiles()
    • set_seed()
    • cheby()
    • random_splits()
    • drop_feature()
    • mask_edge()
    • preprocess_features()
    • compute_ppr()
JittorGeometric
  • Python Module Index

Python Module Index

j
 
j
- jittor_geometric
    jittor_geometric.data
    jittor_geometric.dataloader
    jittor_geometric.datasets
    jittor_geometric.io
    jittor_geometric.nn.aggr
    jittor_geometric.nn.conv
    jittor_geometric.nn.dense
    jittor_geometric.ops
    jittor_geometric.partition
    jittor_geometric.transforms
    jittor_geometric.utils

© Copyright 2025, JittorGeometric Team.

Built with Sphinx using a theme provided by Read the Docs.