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 orderm
- 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