8000 feat: add signatures for `broadcast_shapes` · data-apis/array-api-comparison@ab8b424 · GitHub
[go: up one dir, main page]

Skip to content

Commit ab8b424

Browse files
committed
feat: add signatures for broadcast_shapes
1 parent f61360f commit ab8b424

File tree

1 file changed

+43
-0
lines changed

1 file changed

+43
-0
lines changed
Lines changed: 43 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,43 @@
1+
# broadcast_shapes
2+
3+
## NumPy
4+
5+
```
6+
numpy.broadcast_shapes(*args) → Tuple
7+
```
8+
9+
## CuPy
10+
11+
```
12+
13+
```
14+
15+
## dask.array
16+
17+
Appears to support for not in docs: <https://github.com/dask/dask/issues/11433>
18+
19+
```
20+
dask.array.broadcast_shapes(*args) → Tuple
21+
```
22+
23+
## JAX
24+
25+
```
26+
jax.numpy.broadcast_shapes(*shapes) → Tuple
27+
```
28+
29+
## PyTorch
30+
31+
```
32+
torch.broadcast_shapes(*shapes) → Size
33+
```
34+
35+
## TensorFlow
36+
37+
```
38+
tf.broadcast_static_shape(shape_x, shape_y) → TensorShape
39+
```
40+
41+
```
42+
tf.broadcast_dynamic_shape(shape_x, shape_y) → Tensor
43+
```

0 commit comments

Comments
 (0)
0