Skip to content

ENH: expand_dims tuple axes#414

Draft
ev-br wants to merge 1 commit intodata-apis:masterfrom
ev-br:expand_dims_tuple_axes
Draft

ENH: expand_dims tuple axes#414
ev-br wants to merge 1 commit intodata-apis:masterfrom
ev-br:expand_dims_tuple_axes

Conversation

@ev-br
Copy link
Member

@ev-br ev-br commented Feb 5, 2026

cross-ref data-apis/array-api#988

needs data-apis/array-api-compat#385 for the torch wrapper,

Draft until 2025 is a thing.

cf data-apis#760 for discussion
We test here that expand_dims with multiple axes is equivalent to
expanding axes one by one---the key is that the axes to add need
to be pre-sorted.
@ev-br ev-br changed the title WIP: Expand dims tuple axes ENH: Expand dims tuple axes Feb 5, 2026
@ev-br ev-br changed the title ENH: Expand dims tuple axes ENH: expand_dims tuple axes Feb 5, 2026
@ev-br ev-br force-pushed the expand_dims_tuple_axes branch from f3b67ac to 54e5862 Compare February 5, 2026 19:19
@ev-br ev-br mentioned this pull request Feb 5, 2026
10 tasks
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant