Extend tensor operations

[tsanona]

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.

Changes

Add some new tensor operations for convenience.

Testing

New function argtopk is tested on doc, other functions use already existing functions.

[laggui]

Great addition for the one_hot method, but I would not add argtopk (unnecessary). 👍

Thanks for fixing all the typos at the same time 😄

[laggui]

LGTM! 🚀

Just one minor thing you missed in the book API

/edit: might have to resolve some conflicts from main

[codecov]

Codecov Report

Attention: Patch coverage is 21.05263% with 30 lines in your changes missing coverage. Please review.

Project coverage is 85.29%. Comparing base (d5e8e31) to head (c48037a).
Report is 1 commits behind head on main.

Files with missing lines	Patch %	Lines
crates/burn-tensor/src/tensor/api/check.rs	4.76%	20 Missing ⚠️
crates/burn-tensor/src/tensor/api/int.rs	0.00%	7 Missing ⚠️
crates/burn-tensor/src/tensor/api/numeric.rs	66.66%	3 Missing ⚠️

Additional details and impacted files

@@            Coverage Diff             @@
##             main    #2413      +/-   ##
==========================================
- Coverage   85.32%   85.29%   -0.03%     
==========================================
  Files         792      792              
  Lines      104455   104479      +24     
==========================================
- Hits        89124    89118       -6     
- Misses      15331    15361      +30     

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

[laggui]

Awesome 🙏