DiagQSM#

class tinygp.solvers.quasisep.core.DiagQSM(d: tinygp.helpers.JAXArray)[source]#

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) → 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) → DiagQSM[source]#

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

self_add(other: DiagQSM) → DiagQSM[source]#

The sum of two DiagQSM matrices

self_mul(other: DiagQSM) → DiagQSM[source]#

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[source]#

The matrix transpose as a QSM