DataSet
==================

.. automodule:: torch_tools.datasets._dataset
   :members:

ShapesDataset
==================

.. automodule:: torch_tools.datasets._shapes_dataset
   :members: