WindowDataset/windows function

[NicoZweifel]

Pull Request Template

Checklist

• Confirmed that run-checks all script has been executed.
• Made sure the book is up to date with changes in this PR.

Related Issues/PRs

• WindowDataset/windows #1551

Changes

• Adds WindowDataset and a windows implementation in Dataset.

Testing

• cd crates/burn-dataset && cargo test window

[codecov]

Codecov Report

All modified and coverable lines are covered by tests ✅

Project coverage is 86.40%. Comparing base (2f88548) to head (c6bba94).
Report is 24 commits behind head on main.

Additional details and impacted files

@@           Coverage Diff           @@
##             main    #1553   +/-   ##
=======================================
  Coverage   86.39%   86.40%           
=======================================
  Files         688      689    +1     
  Lines       78675    78704   +29     
=======================================
+ Hits        67974    68002   +28     
- Misses      10701    10702    +1     

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

[NicoZweifel]

@antimora I could add an individual WindowDataset section here if required but I thought for now the crate documentation would be sufficient as it is quite clear what is going on and the api isn't too complex.

[nathanielsimard]

LGTM

[laggui]

Not extremely familiar with time-series tasks but thank you for the pointers in the linked issue! I can understand the use cases better.

Implementation LGTM! Clean, concise and easy to understand with the added tests 🙂