UpperTriQSM#

class tinygp.solvers.quasisep.core.UpperTriQSM(diag: tinygp.solvers.quasisep.core.DiagQSM, upper: tinygp.solvers.quasisep.core.StrictUpperTriQSM)[source]#

Bases: tinygp.solvers.quasisep.core.QSM

A upper triangular quasiseparable matrix

Parameters

  • diag – The diagonal elements.

  • upper – The strictly upper triangular elements.

matmul(x: tinygp.helpers.JAXArray) → tinygp.helpers.JAXArray[source]#

The dot product of this matrix with a dense vector or matrix

Parameters

x (n, ...) – A matrix or vector with leading dimension matching this matrix.

scale(other: tinygp.helpers.JAXArray) → UpperTriQSM[source]#

The multiplication of this matrix times a scalar, as a QSM

property shape: Tuple[int, int]#

The shape of the matrix

solve(y: tinygp.helpers.JAXArray) → tinygp.helpers.JAXArray[source]#

Solve a linear system with this matrix

If this matrix is called U, this solves U @ x = y for x given y, using backward substitution.

Parameters

y (n, ...) – A matrix or vector with leading dimension matching this matrix.

to_dense() → tinygp.helpers.JAXArray#

Render this representation to a dense matrix

This implementation is not optimized and should really only ever be used for testing purposes.

transpose() → LowerTriQSM[source]#

The matrix transpose as a QSM