search

mmcdermott / nested_ragged_tensors

Utilities for efficiently working with, saving, and loading, collections of connected nested ragged tensors in PyTorch
MIT License
7 stars 1 forks source link

`load_slice` should be able to sub-splice more than just through the zeroth dimension. #9

Closed mmcdermott closed 1 week ago

mmcdermott commented 2 months ago

E.g., I should be able to load not just the data for only one patient, but also a subset of their events as needed.