Preserve static shape information in block_diag

[jessegrabowski]

Description

Small enhancement to block_diag. Adds some reasoning about static shape information to make_node, and propagates this information if all input shapes are known.

Related Issue

• Closes #
• Related to #

Checklist

• Checked that the pre-commit linting/style checks pass
• Included tests that prove the fix is effective or that the new feature works
• Added necessary documentation (docstrings and/or example notebooks)
• If you are a pro: each commit corresponds to a relevant logical change

Type of change

• New feature / enhancement
• Bug fix
• Documentation
• Maintenance
• Other (please specify):

---

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

[codecov]

Codecov Report

✅ All modified and coverable lines are covered by tests.
✅ Project coverage is 81.85%. Comparing base (1d82fb4) to head (a3e9f89).
⚠️ Report is 104 commits behind head on main.

Additional details and impacted files

@@            Coverage Diff             @@
##             main    #1529      +/-   ##
==========================================
- Coverage   82.04%   81.85%   -0.19%     
==========================================
  Files         230      230              
  Lines       52346    52522     +176     
  Branches     9212     9345     +133     
==========================================
+ Hits        42947    42992      +45     
  Misses       7095     7095              
- Partials     2304     2435     +131     

Files with missing lines	Coverage Δ
pytensor/tensor/slinalg.py	93.39% <100.00%> (+0.01%)	⬆️

... and 6 files with indirect coverage changes

🚀 New features to boost your workflow:

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

[ricardoV94]

Why is there a distinct BaseBlockDiag, what does it do?

[jessegrabowski]

It holds logic shared by BlockDiag and SparseBlockDiag

[ricardoV94]

Thanks didn't know we have a sparse