Extend chunk to take size as an argument

[theabhirath]

Should close #109. I would've left this to the author of that issue but I need this for some of the Metalhead.jl work on Res2Net.

---

This PR extends chunk to expect an optional kwarg size, which should return chunks of size size. I've rewritten the way chunk(x, n) works since that seemed like a natural extension and tests pass, but I'm not an expert at the rrule stuff so if there's something wrong feedback would be appreciated 😅

[gruberchr]

Adding method chunk(x; size) looks good to me. A test for the new method is still missing. Or is it sufficient to test it implicitly via method chunk(x, n)?
Additionally I added some comments suggesting to use function cld() for simplification.
I'm not familiar with writing rrules as well, therefore I'm not able to check that.

[theabhirath]

Thanks for the comments! I've changed the code to use cld, and I've also added some tests with chunk taking in size

[codecov-commenter]

Codecov Report

Merging #112 (1128724) into main (b29bf41) will decrease coverage by 0.18%.
The diff coverage is 80.00%.

@@            Coverage Diff             @@
##             main     #112      +/-   ##
==========================================
- Coverage   88.55%   88.36%   -0.19%     
==========================================
  Files          13       13              
  Lines         533      533              
==========================================
- Hits          472      471       -1     
- Misses         61       62       +1     

Impacted Files	Coverage Δ
src/utils.jl	90.24% <77.77%> (-0.82%)	⬇️
src/batchview.jl	80.85% <100.00%> (ø)

---

Continue to review full report at Codecov.

Legend - Click here to learn more
Δ = absolute <relative> (impact), ø = not affected, ? = missing data
Powered by Codecov. Last update b29bf41...1128724. Read the comment docs.

[gruberchr]

👍

[theabhirath]

Bump?