DiagQSM

class tinygp.solvers.quasisep.core.DiagQSM(d: JAXArray)

Bases: QSM

A diagonal quasiseparable matrix

Parameters:

d (n,) – The diagonal entries of the matrix as a 1-D array.

matmul(x: tinygp.helpers.JAXArray, *, parallel: bool = False) → tinygp.helpers.JAXArray

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.

• parallel – If True, use a parallel associative-scan algorithm instead of the default sequential scan.

scale(other: tinygp.helpers.JAXArray) → DiagQSM

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

self_add(other: DiagQSM) → DiagQSM

The sum of two DiagQSM matrices

self_mul(other: DiagQSM) → DiagQSM

The elementwise product of two DiagQSM matrices

property shape: tuple[int, int]

The shape of the 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() → DiagQSM

The matrix transpose as a QSM