search

dllllb / pytorch-lifestream

A library built upon PyTorch for building embeddings on discrete event sequences using self-supervision
Apache License 2.0
215 stars 46 forks source link

Add PaddedBatch slicing #131

Open Ilykuleshov opened 9 months ago

Ilykuleshov commented 9 months ago

Add slicing support on PaddedBatch seq features, similar to FeatureDict.seq_indexing. Would return PaddedBatch, changing seq_lens accordingly.