Add prod and prod_dim tensor ops

[antimora]

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

1. Fixes prod and prod_axis tensor operators #526
2. Left over: Implement prod and prod_dim for Autodiff backend #1458
3. Left over: Implement prod and prod_dim for Candle backend #1454
4. Left over: Implement prod and prod_dim for WGPU backend #1461

Changes

1. Default float implementations for prod and prod_dim
2. Ndarray backend
3. Tch backend
4. New tests
5. Update the docs

Testing

1. New unit tests
2. Book render

[codecov]

Codecov Report

Attention: Patch coverage is 24.28571% with 159 lines in your changes are missing coverage. Please review.

Project coverage is 85.70%. Comparing base (80aac1d) to head (6d45239).

Files	Patch %	Lines
crates/burn-fusion/src/ops/int.rs	0.00%	39 Missing ⚠️
crates/burn-tensor/src/tests/ops/aggregation.rs	53.48%	20 Missing ⚠️
crates/burn-ndarray/src/ops/base.rs	0.00%	17 Missing ⚠️
crates/burn-fusion/src/stream/context.rs	0.00%	11 Missing ⚠️
crates/burn-ndarray/src/ops/macros.rs	0.00%	11 Missing ⚠️
crates/burn-tch/src/ops/base.rs	28.57%	10 Missing ⚠️
crates/burn-jit/src/ops/int_ops.rs	0.00%	9 Missing ⚠️
crates/burn-ndarray/src/ops/int_tensor.rs	0.00%	9 Missing ⚠️
crates/burn-autodiff/src/ops/int_tensor.rs	0.00%	6 Missing ⚠️
crates/burn-fusion/src/stream/operation.rs	0.00%	6 Missing ⚠️
... and 4 more

Additional details and impacted files

@@            Coverage Diff             @@
##             main    #1460      +/-   ##
==========================================
- Coverage   85.88%   85.70%   -0.19%     
==========================================
  Files         645      645              
  Lines       71749    71953     +204     
==========================================
+ Hits        61625    61666      +41     
- Misses      10124    10287     +163     

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

[nathanielsimard]

I don't think it's a good idea, since we may have u8 implement the FloatElement with quantization. Is there a reason to add this trait?

[antimora]

Let me review it again. It might have been a left over.

[antimora]

Removed it. It was a left over.

[laggui]

LGTM! I only had a question but that was already addressed by the previous review + modifications.