Backport "Fix 0-dim broadcasting bug" to 0.9

[mtfishman]

No description provided.

[codecov]

Codecov Report

✅ All modified and coverable lines are covered by tests.
✅ Project coverage is 74.88%. Comparing base (31c97f4) to head (44ff62b).
⚠️ Report is 1 commits behind head on release-0.9.

Additional details and impacted files

@@               Coverage Diff               @@
##           release-0.9     #177      +/-   ##
===============================================
- Coverage        75.34%   74.88%   -0.47%     
===============================================
  Files               37       37              
  Lines             2093     2094       +1     
===============================================
- Hits              1577     1568       -9     
- Misses             516      526      +10     

Flag	Coverage Δ
docs	7.17% <0.00%> (-0.01%)	⬇️

Flags with carried forward coverage won't be shown. Click here to find out more.

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

🚀 New features to boost your workflow:

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