There was an error while loading. Please reload this page.
extra_ops.cumsum
axis
The following raises:
import pytensor.tensor as pt import numpy as np x = pt.vector() x_val = np.array([1.0, 2.0, 3.0]) pt.cumsum(x, axis=0).eval({x: x_val}, mode='NUMBA')