dgl.nn

Package for neural network common components.

• NN Modules (PyTorch)
  • Conv Layers
    • GraphConv
    • RelGraphConv
    • TAGConv
    • GATConv
    • EdgeConv
    • SAGEConv
    • SGConv
    • APPNPConv
    • GINConv
    • GatedGraphConv
    • GMMConv
    • ChebConv
    • AGNNConv
    • NNConv
    • AtomicConv
  • Dense Conv Layers
    • DenseGraphConv
    • DenseSAGEConv
    • DenseChebConv
  • Global Pooling Layers
    • SumPooling
    • AvgPooling
    • MaxPooling
    • SortPooling
    • GlobalAttentionPooling
    • Set2Set
    • SetTransformerEncoder
    • SetTransformerDecoder
  • Utility Modules
    • Sequential
    • KNNGraph
    • SegmentedKNNGraph
    • Edge Softmax
• NN Modules (MXNet)
  • Conv Layers
    • GraphConv
    • RelGraphConv
    • TAGConv
    • GATConv
    • EdgeConv
    • SAGEConv
    • SGConv
    • APPNPConv
    • GINConv
    • GatedGraphConv
    • GMMConv
    • ChebConv
    • AGNNConv
    • NNConv
  • Dense Conv Layers
    • DenseGraphConv
    • DenseSAGEConv
    • DenseChebConv
  • Global Pooling Layers
    • SumPooling
    • AvgPooling
    • MaxPooling
    • SortPooling
    • GlobalAttentionPooling
    • Set2Set
  • Utility Modules
    • Sequential
    • Edge Softmax
• NN Modules (Tensorflow)
  • Conv Layers
    • GraphConv
    • RelGraphConv
    • GATConv
    • SAGEConv
    • SGConv
    • APPNPConv
    • GINConv
  • Global Pooling Layers
    • SumPooling
    • AvgPooling
    • MaxPooling
    • SortPooling
    • GlobalAttentionPooling
  • Utility Modules
    • Edge Softmax