spine.data.batch.TensorBatch
============================

.. currentmodule:: spine.data.batch

.. autoclass:: TensorBatch

   
   .. automethod:: __init__

   
   .. rubric:: Methods

   .. autosummary::
   
      ~TensorBatch.__init__
      ~TensorBatch.apply_mask
      ~TensorBatch.from_list
      ~TensorBatch.get_counts
      ~TensorBatch.get_edges
      ~TensorBatch.merge
      ~TensorBatch.split
      ~TensorBatch.to_cm
      ~TensorBatch.to_numpy
      ~TensorBatch.to_px
      ~TensorBatch.to_tensor
   
   

   
   
   .. rubric:: Attributes

   .. autosummary::
   
      ~TensorBatch.batch_ids
      ~TensorBatch.shape
      ~TensorBatch.splits
      ~TensorBatch.tensor
      ~TensorBatch.data
      ~TensorBatch.counts
      ~TensorBatch.edges
      ~TensorBatch.batch_size
      ~TensorBatch.has_batch_col
      ~TensorBatch.coord_cols