Linear Algebra Functions

Array API specification for linear algebra functions.

A conforming implementation of the array API standard must provide and support the following functions.

Objects in API

matmul(x1, x2, /)

Computes the matrix product.

matrix_transpose(x, /)

Transposes a matrix (or a stack of matrices) x.

tensordot(x1, x2, /, *, axes=2)

Returns a tensor contraction of x1 and x2 over specific axes.

vecdot(x1, x2, /, *, axis=-1)

Computes the (vector) dot product of two arrays.