Skip to content

extra_ops.cumsum raises in numba backend on 1d input when the axis argument is specified #1689

@jessegrabowski

Description

@jessegrabowski

Description

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')

Metadata

Metadata

Assignees

No one assigned

    Labels

    bugSomething isn't workingnumba

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions