noether.data.pipeline.collators.sparse_tensor_offset

Classes

SparseTensorOffsetCollator

Collates sparse tensors by concatenating them along the first axis and creating an offset tensor that maps

Module Contents

class noether.data.pipeline.collators.sparse_tensor_offset.SparseTensorOffsetCollator(item, offset_key)

Bases: noether.data.pipeline.collator.Collator

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.

Parameters:
item
offset_key