emmi_inference.models.pipelines.sparse_tensor_offset
====================================================

.. py:module:: emmi_inference.models.pipelines.sparse_tensor_offset

.. autoapi-nested-parse::

   WARNING
   This file is a 1:1 duplicate from the ``tutorial`` folder.

   It is here to avoid installation of a tutorial as a package and keep it isolated.



Classes
-------

.. autoapisummary::

   emmi_inference.models.pipelines.sparse_tensor_offset.SparseTensorOffsetCollator


Module Contents
---------------

.. py:class:: SparseTensorOffsetCollator(item, offset_key)

   Collates sparse tensors by concatenating them along the first axis and creating an offset tensor that maps
   each sample to its respective index in the batch.


   .. py:attribute:: item


   .. py:attribute:: offset_key