Fix/fuse select

[nathanielsimard]

Fix fuse select with vectorization on another dimension other than the last one.

[codecov]

Codecov Report

Attention: Patch coverage is 20.72072% with 88 lines in your changes missing coverage. Please review.

Project coverage is 81.66%. Comparing base (e23c8ef) to head (3d0c1a7).
Report is 1 commits behind head on main.

Files with missing lines	Patch %	Lines
crates/burn-jit/src/fusion/on_write/kernel.rs	8.33%	88 Missing ⚠️

Additional details and impacted files

@@            Coverage Diff             @@
##             main    #2804      +/-   ##
==========================================
- Coverage   81.70%   81.66%   -0.04%     
==========================================
  Files         851      851              
  Lines      113346   113413      +67     
==========================================
+ Hits        92610    92620      +10     
- Misses      20736    20793      +57     

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