stgraph.nn.pytorch package

Subpackages

• stgraph.nn.pytorch.static package
  • Submodules
  • stgraph.nn.pytorch.static.gat_conv module
    • GATConv
      • GATConv.forward()
      • GATConv.reset_parameters()
      • GATConv.training
  • stgraph.nn.pytorch.static.gcn_conv module
    • GCNConv
      • GCNConv.forward()
      • GCNConv.reset_parameters()
      • GCNConv.training
  • Module contents
• stgraph.nn.pytorch.temporal package
  • Submodules
  • stgraph.nn.pytorch.temporal.tgcn module
    • TGCN
      • TGCN.forward()
      • TGCN.training
  • Module contents

Module contents

Vertex-centric implementation for state-of-the-art Graph Neural Networks for PyTorch specific backend.