We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 9bce442 commit c746621Copy full SHA for c746621
array_api_compat/cupy/_aliases.py
@@ -139,6 +139,11 @@ def take_along_axis(x: Array, indices: Array, /, *, axis: int = -1) -> Array:
139
return cp.take_along_axis(x, indices, axis=axis)
140
141
142
+# https://github.com/cupy/cupy/pull/9582
143
+def broadcast_arrays(*arrays: Array) -> tuple[Array, ...]:
144
+ return tuple(cp.broadcast_arrays(*arrays))
145
+
146
147
# These functions are completely new here. If the library already has them
148
# (i.e., numpy 2.0), use the library version instead of our wrapper.
149
if hasattr(cp, 'vecdot'):
@@ -161,7 +166,8 @@ def take_along_axis(x: Array, indices: Array, /, *, axis: int = -1) -> Array:
161
166
'atan2', 'atanh', 'bitwise_left_shift',
162
167
'bitwise_invert', 'bitwise_right_shift',
163
168
'bool', 'concat', 'count_nonzero', 'pow', 'sign',
164
- 'ceil', 'floor', 'trunc', 'take_along_axis']
169
+ 'ceil', 'floor', 'trunc', 'take_along_axis',
170
+ 'broadcast_arrays',]
165
171
172
173
def __dir__() -> list[str]:
0 commit comments