PyTorch Geometric Temporal
==================

.. contents:: Contents
    :local:

Recurrent Graph Convolutional Layers
--------------

.. automodule:: torch_geometric_temporal.nn.recurrent.gconv_gru
    :members:
    :undoc-members:

.. automodule:: torch_geometric_temporal.nn.recurrent.gconv_lstm
    :members:
    :undoc-members:

.. automodule:: torch_geometric_temporal.nn.recurrent.gc_lstm
    :members:
    :undoc-members:

.. automodule:: torch_geometric_temporal.nn.recurrent.lrgcn
    :members:
    :undoc-members:

.. automodule:: torch_geometric_temporal.nn.recurrent.dygrae
    :members:
    :undoc-members:

.. automodule:: torch_geometric_temporal.nn.recurrent.evolvegcnh
    :members:
    :undoc-members:

.. automodule:: torch_geometric_temporal.nn.recurrent.evolvegcno
    :members:
    :undoc-members:
    
.. automodule:: torch_geometric_temporal.nn.recurrent.temporalgcn
    :members:
    :undoc-members:
    
.. automodule:: torch_geometric_temporal.nn.recurrent.attentiontemporalgcn
    :members:
    :undoc-members:
    
.. automodule:: torch_geometric_temporal.nn.recurrent.mpnn_lstm
    :members:
    :undoc-members:

.. automodule:: torch_geometric_temporal.nn.recurrent.dcrnn
    :members:
    :undoc-members:
    :exclude-members: DConv
    
.. automodule:: torch_geometric_temporal.nn.recurrent.agcrn
    :members:
    :undoc-members:
    :exclude-members: AVWGCN

Temporal Graph Attention Layers
--------------

.. automodule:: torch_geometric_temporal.nn.attention.stgcn
    :members:
    :undoc-members:
    :exclude-members: TemporalConv

.. automodule:: torch_geometric_temporal.nn.attention.astgcn
    :members:
    :undoc-members:
    :exclude-members:  SpatialAttention, ASTGCNBlock, TemporalAttention, ChebConvAttention
    
.. automodule:: torch_geometric_temporal.nn.attention.mstgcn
    :members:
    :undoc-members:
    :exclude-members: MSTGCNBlock

.. automodule:: torch_geometric_temporal.nn.attention.gman
    :members:
    :undoc-members:
    :exclude-members: Conv2D, FullyConnected, SpatioTemporalEmbedding, SpatialAttention, TemporalAttention, GatedFusion, SpatioTemporalAttentionBlock, TransformAttention
    
.. automodule:: torch_geometric_temporal.nn.attention.mtgnn
    :members:
    :undoc-members:
    :exclude-members: Linear, MixProp, DilatedInception, LayerNormalization, MTGNNLayer

.. automodule:: torch_geometric_temporal.nn.attention.tsagcn
    :members:
    :undoc-members:
    :exclude-members: UnitGCN, UnitTCN, bn_init, conv_init, conv_branch_init, GraphAAGCN
    
    
.. automodule:: torch_geometric_temporal.nn.attention.dnntsp
    :members:
    :undoc-members:
    :exclude-members: WeightedGCNBlock, AggregateTemporalNodeFeatures, GlobalGatedUpdater, MaskedSelfAttention


Auxiliary Graph Convolutional Layers
--------------

.. automodule:: torch_geometric_temporal.nn.attention.stgcn
    :members:
    :undoc-members:
    :exclude-members: STConv

.. automodule:: torch_geometric_temporal.nn.recurrent.dcrnn
    :members:
    :undoc-members:
    :exclude-members: DCRNN
    
.. automodule:: torch_geometric_temporal.nn.attention.astgcn
    :members:
    :undoc-members:
    :exclude-members:  SpatialAttention, ASTGCNBlock, TemporalAttention, ASTGCN
    
.. automodule:: torch_geometric_temporal.nn.recurrent.agcrn
    :members:
    :undoc-members:
    :exclude-members: AGCRN

.. automodule:: torch_geometric_temporal.nn.attention.tsagcn
    :members:
    :undoc-members:
    :exclude-members: bn_init, conv_init, conv_branch_init, GraphAAGCN, AAGCN


Heterogeneous Graph Convolutional Layers
--------------

.. automodule:: torch_geometric_temporal.nn.hetero.heterogclstm
    :members:
    :undoc-members:
