Support signed value for Tensor::arange

[Nikaidou-Shinku]

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

#1234

Changes

Make Tensor::arange and all internal methods accept a Range<i64> instead of Range<usize> to support ranges of signed value.

Testing

bash ./run-checks.sh all was passed.

[codecov]

Codecov Report

Attention: 12 lines in your changes are missing coverage. Please review.

Comparison is base (9df2071) 84.41% compared to head (a766252) 84.38%.
Report is 9 commits behind head on main.

Files	Patch %	Lines
burn-tch/src/ops/int_tensor.rs	0.00%	9 Missing ⚠️
burn-autodiff/src/ops/int_tensor.rs	0.00%	3 Missing ⚠️

Additional details and impacted files

@@            Coverage Diff             @@
##             main    #1238      +/-   ##
==========================================
- Coverage   84.41%   84.38%   -0.04%     
==========================================
  Files         549      550       +1     
  Lines       61952    62206     +254     
==========================================
+ Hits        52295    52490     +195     
- Misses       9657     9716      +59     

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

[nathanielsimard]

I saw some naming problems related to arange, I think we could fix this within this PR.

[antimora]

1. A new unit test should be added to test all new edge cases (negatives and 0, overflow range).

2. Extra caution when casting usize to i64 because it overflow.

[nathanielsimard]

1. A new unit test should be added to test all new edge cases (negatives and 0, overflow range).
2. Extra caution when casting usize to i64 because it overflow.

I think we should check if we can replace the usize by i64 instead of casting in most places.

[Nikaidou-Shinku]

I moved the functions about arange to int_tensor.rs, and add some test cases for 0 and negative numbers.
Regarding overflow, I don't have any good ideas. I noticed that for most cases where arange is used in the test cases, the range is taken from the shape of tensors. In this case, when an overflow occurs, I don't know what else to do but crash the program.

[antimora]

LGTM