8000 Add signatures for `repeat` · data-apis/array-api-comparison@e38b7d4 · GitHub
[go: up one dir, main page]

Skip to content

Commit e38b7d4

Browse files
committed
Add signatures for repeat
1 parent 2f1ccfc commit e38b7d4

File tree

1 file changed

+47
-0
lines changed

1 file changed

+47
-0
lines changed

signatures/manipulation/repeat.md

Lines changed: 47 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,47 @@
1+
# repeat
2+
3+
## NumPy
4+
5+
```
6+
numpy.repeat(a, repeats, axis=None) → ndarray
7+
```
8+
9+
## CuPy
10+
11+
```
12+
cupy.repeat(a, repeats, axis=None) → ndarray
13+
```
14+
15+
## dask.array
16+
17+
```
18+
dask.array.repeat(a, repeats, axis=None) → ndarray
19+
```
20+
21+
## JAX
22+
23+
```
24+
jax.numpy.repeat(a, repeats, axis=None, *, total_repeat_length=None) → ndarray
25+
```
26+
27+
Note: requires `total_repeat_length` to be provided in order to be compilable.
28+
29+
## MXNet
30+
31+
```
32+
np.repeat(a, repeats, axis=None) → ndarray
33+
```
34+
35+
## PyTorch
36+
37+
```
38+
torch.repeat_interleave(input, repeats, dim=None, *, output_size=None) → Tensor
39+
```
40+
41+
Note: `torch.Tensor.repeat` behaves similar to NumPy `tile`.
42+
43+
## TensorFlow
44+
45+
```
46+
tf.repeat(input, repeats, axis=None, name=None) → Tensor
47+
```

0 commit comments

Comments
 (0)
0