pytensor
pytensor copied to clipboard
Make Elemwise a subclass of Blockwise
Description
All general Blockwise methods (infer_shape, grad, ...) work for Elemwise which is just a trivial Blockwise function with signature "(),()...()->(),...,()` (that is one entry for each input and output).
We should make Elemwise a subclass of Blockwise and remove any redundant methods.