Weight initialisation functions

Weight initialisation
=====================

.. automodule:: torch_tools.weight_init
   :members: