****************
PyTorch Blocks
****************

Here can be found all blocks implemented in PyTorch for the architecture's implementation in that framework.

.. toctree::
    :caption: PyTorch Blocks
    :maxdepth: 1

.. autosummary::
    TSFEDL.blocks_pytorch.ConvBlockYiboGao
    TSFEDL.blocks_pytorch.AttentionBranchYiboGao
    TSFEDL.blocks_pytorch.RTABlock
    TSFEDL.blocks_pytorch.SqueezeAndExcitationModule
    TSFEDL.blocks_pytorch.DenseNetTransitionBlock
    TSFEDL.blocks_pytorch.DenseNetConvBlock
    TSFEDL.blocks_pytorch.DenseNetDenseBlock
    TSFEDL.blocks_pytorch.SpatialAttentionBlockZhangJin
    TSFEDL.blocks_pytorch.TemporalAttentionBlockZhangJin

.. autofunction:: TSFEDL.blocks_pytorch.ConvBlockYiboGao
.. autofunction:: TSFEDL.blocks_pytorch.AttentionBranchYiboGao
.. autofunction:: TSFEDL.blocks_pytorch.RTABlock
.. autofunction:: TSFEDL.blocks_pytorch.SqueezeAndExcitationModule
.. autofunction:: TSFEDL.blocks_pytorch.DenseNetTransitionBlock
.. autofunction:: TSFEDL.blocks_pytorch.DenseNetConvBlock
.. autofunction:: TSFEDL.blocks_pytorch.DenseNetDenseBlock
.. autofunction:: TSFEDL.blocks_pytorch.SpatialAttentionBlockZhangJin
.. autofunction:: TSFEDL.blocks_pytorch.TemporalAttentionBlockZhangJin