pytensor icon indicating copy to clipboard operation
pytensor copied to clipboard

Rewrite scalar dot as multiplication #1205

Open SwarnimShekhar opened this issue 9 months ago • 0 comments

Description

  • Optimized local_blockwise_dot_to_mul to handle (1,1) @ (1,1) cases as direct elementwise multiplication.
  • Prevents unnecessary reshaping for scalar-like tensors.
  • Ensures consistency with other elementwise multiplication cases.

Related Issue

  • [ ] Closes #
  • [ ] Related to #

Checklist

Type of change

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

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

SwarnimShekhar avatar Mar 04 '25 16:03 SwarnimShekhar