NeuroSurgeon.Masking package
============================

Submodules
----------

``Mask Layer``
*****************************

.. automodule:: NeuroSurgeon.Masking.mask_layer
   :members:
   :undoc-members:
   :show-inheritance:

``Continuous Sparsification``
*****************************

.. automodule:: NeuroSurgeon.Masking.contsparse_layer
   :members:
   :undoc-members:
   :show-inheritance:

``Hard Concrete Masking``
*****************************


.. automodule:: NeuroSurgeon.Masking.hardconcrete_layer
   :members:
   :undoc-members:
   :show-inheritance:

``Magnitude Pruning``
*****************************

.. automodule:: NeuroSurgeon.Masking.magprune_layer
   :members:
   :undoc-members:
   :show-inheritance:

Module contents
---------------

.. automodule:: NeuroSurgeon.Masking
   :members:
   :undoc-members:
   :show-inheritance: