Rewrite more cases of Blockwise IncSubtensor

[ricardoV94]

Spin-off from #1558 with the rewrite to remove Blockwise IncSubtensor extended to basic IncSubtensor (before only covered AdvancedIncSubtensor) and to batch indices. This shows up frequently if building vectorized jacobians so we want to make sure it's optimized away.

We could perhaps do the rewrite eagerly when we call vectorize_node, but since the logic is pretty complex I decided to keep it in a rewrite. The graph with Blockwise is still readable, so it's merely a matter of performance / enabling other Subtensor rewrites (including inplace!)

PS: We should be able to reuse the arange logic to rewrite away blockwise Subtensor with batch indices and AdvancedSubtensor.

Except for batched slices, which may not always be vectorizable in a "square manner", we shouldn't ever end up with a Blockwise of a subtensor (or a subtensor update) in the final graph. Numpy indexing is flexible enough to cover any vectorization case, it's just not trivial to write it :D

---

📚 Documentation preview 📚: https://pytensor--1560.org.readthedocs.build/en/1560/

[codecov]

Codecov Report

❌ Patch coverage is 98.78049% with 1 line in your changes missing coverage. Please review.
✅ Project coverage is 81.54%. Comparing base (b55c473) to head (9e9d8ed).

Files with missing lines	Patch %	Lines
pytensor/tensor/rewriting/subtensor.py	98.64%	0 Missing and 1 partial ⚠️

Additional details and impacted files

@@            Coverage Diff             @@
##             main    #1560      +/-   ##
==========================================
+ Coverage   81.53%   81.54%   +0.01%     
==========================================
  Files         230      230              
  Lines       53079    53099      +20     
  Branches     9425     9432       +7     
==========================================
+ Hits        43279    43302      +23     
+ Misses       7365     7364       -1     
+ Partials     2435     2433       -2     

Files with missing lines	Coverage Δ
pytensor/tensor/rewriting/blockwise.py	96.24% <ø> (-0.32%)	⬇️
pytensor/tensor/subtensor.py	89.89% <100.00%> (-0.12%)	⬇️
pytensor/tensor/rewriting/subtensor.py	91.02% <98.64%> (+0.74%)	⬆️

... and 1 file with indirect coverage changes

🚀 New features to boost your workflow:

• ❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.