8000 Add matmul and remove extra parenthesis · pmeier/array-api-comparison@cbc43f2 · GitHub
[go: up one dir, main page]

Skip to content

Commit cbc43f2

Browse files
committed
Add matmul and remove extra parenthesis
1 parent 489ad44 commit cbc43f2

File tree

2 files changed

+46
-2
lines changed

2 files changed

+46
-2
lines changed

signatures/linalg/matmul.md

Lines changed: 44 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,44 @@
1+
# matmul
2+
3+
## NumPy
4+
5+
```
6+
numpy.linalg.matmul(x1, x2, /, out=None, *, casting='same_kind', order='K', dtype=None, subok=True[, signature, extobj]) → ndarray
7+
```
8+
9+
## CuPy
10+
11+
```
12+
cupy.matmul(a, b, out=None) → ndarray
13+
```
14+
15+
## dask.array
16+
17+
```
18+
dask.array.matmul(x1, x2, /, out=None, *, casting='same_kind', order='K', dtype=None, subok=True[, signature, extobj]) → ndarray
19+
```
20+
21+
## JAX
22+
23+
```
24+
jax.numpy.matmul(a, b, *, precision=None) → ndarray
25+
```
26+
27+
## MXNet
28+
29+
```
30+
np.linalg.matmul(a, b, out=None) → ndarray
31+
```
32+
33+
## PyTorch
34+
35+
```
36+
torch.matmul(input, other, *, out=None) → Tensor
37+
```
38+
39+
## TensorFlow
40+
41+
```
42+
tf.linalg.matmul(a, b, transpose_a=False, transpose_b=False, adjoint_a=False, adjoint_b=False,
43+
a_is_sparse=False, b_is_sparse=False, name=None) → Tensor
44+
```

signatures/linalg/matrix_rank.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -11,10 +11,10 @@ tol: `S.max() * max(M.shape) * eps`
1111
## CuPy
1212

1313
```
14-
cupy.linalg.matrix_rank(M, tol=None)) → ndarray
14+
cupy.linalg.matrix_rank(M, tol=None) → ndarray
1515
```
1616

17-
`tol` can only be float, not array-like.
17+
`tol` can only be float, not array-like. Does not support stacks.
1818

1919
## dask.array
2020

0 commit comments

Comments
 (0)
0