binius_math

Module fold

Source
Expand description

Algorithms for matrix multiplications and generalized tensor contractions.

Functionsยง

fold_cols
Computes a linear combination of the columns of a matrix.
fold_rows
Computes a linear combination of the rows of a matrix.