Implement `infer_shape` automatically from `gufunc_signature`

Open ricardoV94 opened this issue 9 months ago • 0 comments

For Ops with a gufunc_signature, we can automate infer_shape implementation:

https://github.com/pymc-devs/pytensor/blob/e036caf934e177c72053a40ffbac258c9c26d5ac/pytensor/tensor/slinalg.py#L28-L53

We actually already do it for the Blockwise Wrapper:

https://github.com/pymc-devs/pytensor/blob/3cdcfde4155b5aa8d4ac29fd7f07fe21f776a712/pytensor/tensor/blockwise.py#L208-L210

Feb 28 '25 10:02 ricardoV94