Add dim checks on output rank for unsqueeze_dim and stack

[laggui]

Checklist

• Confirmed that run-checks all script has been executed.

Related Issues/PRs

Fixes #2329 (better help message)

Changes

Added checks for output rank with unsqueeze_dim and stack checks.

[codecov]

Codecov Report

Attention: Patch coverage is 67.85714% with 9 lines in your changes missing coverage. Please review.

Project coverage is 85.36%. Comparing base (99d9fa2) to head (7b9f59c).
Report is 1 commits behind head on main.

Files with missing lines	Patch %	Lines
crates/burn-tensor/src/tensor/api/check.rs	65.38%	9 Missing ⚠️

Additional details and impacted files

@@            Coverage Diff             @@
##             main    #2331      +/-   ##
==========================================
- Coverage   85.37%   85.36%   -0.01%     
==========================================
  Files         768      769       +1     
  Lines       98673    98692      +19     
==========================================
+ Hits        84238    84249      +11     
- Misses      14435    14443       +8     

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