lenskit.math.solve#

Efficient solver routines.

Functions#

solve_cholesky(A, y)

Solve the system \(A\mathbf{x}=\mathbf{y}\) for \(\mathbf{x}\)

Module Contents#

lenskit.math.solve.solve_cholesky(A, y)#

Solve the system \(A\mathbf{x}=\mathbf{y}\) for \(\mathbf{x}\) with Cholesky decomposition.

This wraps torch.linalg.cholesky_ex() and torch.cholesky_solve() in an easier-to-use interface with error checking.

Parameters:
Returns:

the solution \(\mathbf{x}\)

Return type:

lenskit.data.types.NPVector

Exported Aliases#

lenskit.math.solve.NPMatrix#

Re-exported alias for lenskit.data.types.NPMatrix.

lenskit.math.solve.NPVector#

Re-exported alias for lenskit.data.types.NPVector.