summaryrefslogtreecommitdiff
path: root/numpy/array_api/_manipulation_functions.py
Commit message (Collapse)AuthorAgeFilesLines
* DOC: Some updates to the array_api compat document (#22747)Aaron Meurer2022-12-061-0/+1
| | | | | | | * Add reshape differences to the array API compat document * Add an item to the array API compat document about reverse broadcasting * Make some wording easier to read
* ENH: Updates to numpy.array_api (#19937)Aaron Meurer2021-09-251-0/+11
| | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | * Add __index__ to array_api and update __int__, __bool__, and __float__ The spec specifies that they should only work on arrays with corresponding dtypes. __index__ is new in the spec since the initial PR, and works identically to np.array.__index__. * Add the to_device method to the array_api This method is new since #18585. It does nothing in NumPy since NumPy does not support non-CPU devices. * Update transpose methods in the array_api transpose() was renamed to matrix_transpose() and now operates on stacks of matrices. A function to permute dimensions will be added once it is finalized in the spec. The attribute mT was added and the T attribute was updated to only operate on 2-dimensional arrays as per the spec. * Restrict input dtypes in the array API statistical functions * Add the dtype parameter to the array API sum() and prod() * Add the function permute_dims() to the array_api namespace permute_dims() is the replacement for transpose(), which was split into permute_dims() and matrix_transpose(). * Add tril and triu to the array API namespace * Fix the array_api Array.__repr__ to indent the array properly * Make the Device type in the array_api just accept the string "cpu"
* Run (selective) black on the array_api submoduleAaron Meurer2021-08-061-2/+16
| | | | | | I've omitted a few changes from black that messed up the readability of some complicated if statements that were organized logically line-by-line, and some changes that use unnecessary operator spacing.
* Make the axis argument to squeeze() in the array_api module positional-onlyAaron Meurer2021-08-061-1/+1
| | | | See data-apis/array-api#100.
* Fix casting for the array API concat() and stack()Aaron Meurer2021-08-041-4/+5
|
* Rename numpy._array_api to numpy.array_apiAaron Meurer2021-08-041-0/+71
Instead of the leading underscore, the experimentalness of the module will be indicated by omitting a warning on import. That we, we do not have to change the API from underscore to no underscore when the module is no longer experimental.