Some random PyTorch utility functions

Torch utils
===========

.. automodule:: torch_tools.torch_utils
   :members: