-
Notifications
You must be signed in to change notification settings - Fork 29
repeat with axis=None
repeats flattened array
#1427
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Conversation
Adds tests for new functionality
Array API standard conformance tests for dpctl=0.15.1dev0=py310ha25a700_2 ran successfully. |
axis=None
repeats flattened array
axis=None
repeats flattened arrayaxis=None
repeats flattened array
Docstring pertaining to description of |
Docstring has been adjusted to reflect changes to `axis` as well as new `repeats` types Corrected a bug in the behavior of `repeat` for size 1 `repeats` Python sequences
View rendered docs @ https://intelpython.github.io/dpctl/pulls/1427/index.html |
Array API standard conformance tests for dpctl=0.15.1dev0=py310ha25a700_3 ran successfully. |
Array API standard conformance tests for dpctl=0.15.1dev0=py310ha25a700_4 ran successfully. |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM! Thank you @ndgrigorian
This PR permits
axis=None
for any input torepeat
.To do so, the previously implemented kernels for repeat were tweaked to permit this behavior, and overloads of
py_repeat_by_sequence
andpy_repeat_by_scalar
withoutaxis
arguments were implemented. These share an overloaded Python binding, with and without anaxis
argument.