Skip to content
New issue

Have a question about this project? # for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “#”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? # to your account

squeeze accepts axis=None? #62

Open
mdhaber opened this issue Sep 20, 2024 · 0 comments
Open

squeeze accepts axis=None? #62

mdhaber opened this issue Sep 20, 2024 · 0 comments

Comments

@mdhaber
Copy link

mdhaber commented Sep 20, 2024

It looks like axis=None is not in the standard for squeeze, yet it is accepted by array_api_strict.

import array_api_strict as xp
x = xp.reshape(xp.arange(12), (3, 4))[..., xp.newaxis]
xp.squeeze(x, axis=None)
# Array([[ 0,  1,  2,  3],
#        [ 4,  5,  6,  7],
#        [ 8,  9, 10, 11]], dtype=array_api_strict.int64)
# for free to join this conversation on GitHub. Already have an account? # to comment
Projects
None yet
Development

No branches or pull requests

2 participants