Linear algebra functionsΒΆ

matmul

Computes the matrix product.

matrix_transpose

Transposes the innermost two dimensions of x, where x is a 2-dimensional matrix or a stack of 2-dimensional matrices.

tensordot

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

vecdot

Computes the (vector) dot product of two arrays.