pytensor
pytensor copied to clipboard
Add `infer_shape` method to Op base class?
Description
For discoverability.
We can raise NotImplementedError
by default.
Or create a subclass of ShapedOp?