:github_url: https://github.com/taipingeric/fusionlab

.. _layers:

Layers
==============
Custom layers

.. automodule:: fusionlab.layers
.. currentmodule:: fusionlab.layers

N-dimensional layer
--------

`Convolution ND`
~~~~~~~~~~
.. autoclass:: ConvND
    :members:

`Transposed Convolution ND`
~~~~~~~~~~
.. autoclass:: ConvT
    :members:

`Upsample ND`
~~~~~~~~~~
.. autoclass:: Upsample
    :members:

`BatchNorm ND`
~~~~~~~~~~
.. autoclass:: BatchNorm
    :members:

`InstanceNorm ND`
~~~~~~~~~~
.. autoclass:: InstanceNorm
    :members:

`Maxpool ND`
~~~~~~~~~~
.. autoclass:: MaxPool
    :members:

`AvgPool ND`
~~~~~~~~~~
.. autoclass:: AvgPool
    :members:

`Global Max Pool (Adaptive Max Pool) ND`
~~~~~~~~~~
.. autoclass:: AdaptiveMaxPool
    :members:

`Global Avg Pool(Adaptive Avg Pool) ND`
~~~~~~~~~~
.. autoclass:: AdaptiveAvgPool
    :members:

`Replication Padding ND`
~~~~~~~~~~
.. autoclass:: ReplicationPad
    :members:

`Constant Padding ND`
~~~~~~~~~~
.. autoclass:: ConstantPad
    :members:

`Conv|Norm|Act`
~~~~~~~~~~
.. autoclass:: ConvNormAct
    :members:

`Rearrange`
~~~~~~~~~~
.. autoclass:: Rearrange
    :members:

`Patch Embedding`
~~~~~~~~~~
.. autoclass:: PatchEmbedding
    :members:

`Squeeze-Excitation`
~~~~~~~~~~
.. autoclass:: SEModule
    :members:
.. autoclass:: TFSEModule
    :members: