Skip to content

Latest commit

 

History

History
24 lines (19 loc) · 617 Bytes

norm.rst

File metadata and controls

24 lines (19 loc) · 617 Bytes

torchdeq.norm

.. automodule:: torchdeq.norm
   :no-members:

Norm Function

.. autofunction:: torchdeq.norm.apply_norm
.. autofunction:: torchdeq.norm.reset_norm
.. autofunction:: torchdeq.norm.remove_norm
.. autofunction:: torchdeq.norm.register_norm
.. autofunction:: torchdeq.norm.register_norm_module


Normalization

.. autoclass:: torchdeq.norm.weight_norm.WeightNorm
   :members:
   :exclude-members: overwrite_kwargs

.. autoclass:: torchdeq.norm.spectral_norm.SpectralNorm
   :members:
   :exclude-members: overwrite_kwargs