.. module:: k1lib.nn

k1lib.nn module
---------------

Some nice utils to complement :mod:`torch.nn`

.. autoclass:: k1lib.nn.Lambda
   :members: __init__
   :show-inheritance:

.. autoclass:: k1lib.nn.Identity
   :show-inheritance: