Skip to content

Remove symbolic axis from join/split#2144

Open
jessegrabowski wants to merge 4 commits into
pymc-devs:mainfrom
jessegrabowski:remove-symbolic-axis
Open

Remove symbolic axis from join/split#2144
jessegrabowski wants to merge 4 commits into
pymc-devs:mainfrom
jessegrabowski:remove-symbolic-axis

Conversation

@jessegrabowski
Copy link
Copy Markdown
Member

@jessegrabowski jessegrabowski commented May 18, 2026

Closes #1528

Removes the symbolic axis argument from join/split. Axis is now provided at Op construction time, so API goes from:

op = Join()(axis, *tensors)

to:

op = Join(axis)(*tensors)

Users who are using the public entrypoints (pt.concatenate, pt.stack, etc) should not see any change in behavior. Downstreams that use Join directly will have to update, but this should be rare.

Move axis from a symbolic Apply-node input to a __props__ property.
Breaking change: Apply-node layout differs and c_code cache versions
are bumped, so old pickled graphs will not load.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Deprecate symbolic Join/Split axis

2 participants