implemented mini-batch extraction with with statement for PyTorch

[LukasMut]

• implemented mini-batch extraction using a with statement to add hooks at the start of extraction and remove them in the end (this is only relevant for PyTorch models)

[codecov[bot]]

Codecov Report

All modified and coverable lines are covered by tests :white_check_mark:

Project coverage is 74.34%. Comparing base (4ceafff) to head (5a74002).

:exclamation: Current head 5a74002 differs from pull request most recent head 2bf4edb. Consider uploading reports for the commit 2bf4edb to get more accurate results

Additional details and impacted files

```diff @@ Coverage Diff @@ ## master #160 +/- ## ========================================== + Coverage 74.12% 74.34% +0.22% ========================================== Files 31 31 Lines 1631 1645 +14 Branches 224 224 ========================================== + Hits 1209 1223 +14 Misses 356 356 Partials 66 66 ``` | [Flag](https://app.codecov.io/gh/ViCCo-Group/thingsvision/pull/160/flags?src=pr&el=flags&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=ViCCo-Group) | Coverage Δ | | |---|---|---| | [unittests](https://app.codecov.io/gh/ViCCo-Group/thingsvision/pull/160/flags?src=pr&el=flag&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=ViCCo-Group) | `74.34% <100.00%> (+0.22%)` | :arrow_up: | Flags with carried forward coverage won't be shown. [Click here](https://docs.codecov.io/docs/carryforward-flags?utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=ViCCo-Group#carryforward-flags-in-the-pull-request-comment) to find out more.

:umbrella: View full report in Codecov by Sentry.
:loudspeaker: Have feedback on the report? Share it here.