pytensor icon indicating copy to clipboard operation
pytensor copied to clipboard

Make Elemwise a subclass of Blockwise

Open ricardoV94 opened this issue 10 months ago • 0 comments

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.

ricardoV94 avatar Feb 16 '25 20:02 ricardoV94