GeneralQSM#

class tinygp.solvers.quasisep.general.GeneralQSM(pl: tinygp.helpers.JAXArray, ql: tinygp.helpers.JAXArray, pu: tinygp.helpers.JAXArray, qu: tinygp.helpers.JAXArray, a: tinygp.helpers.JAXArray, idx: tinygp.helpers.JAXArray)[source]#

Bases: object

A rectangular (n1,n2) quasiseparable matrix with order m

Parameters

  • pl (n1, m) – The lower left quasiseparable vectors.

  • ql (n2, m) – The lower right quasiseparable vectors.

  • pu (n2, m) – The upper right quasiseparable vectors.

  • qu (n1, m) – The upper left quasiseparable vectors.

  • a (n1, m, m) – The transition matrices.

  • idx (n1,) – The indices of the diagonal.

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

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

Parameters

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

property shape: Tuple[int, int]#

The shape of the matrix