probabilistic-numerics / probnum

Probabilistic Numerics in Python.
http://probnum.org
MIT License
438 stars 57 forks source link

Implement efficient (advanced) indexing, slicing, and masking on `LinearOperator`s and `Normal` random variables #193

Open marvinpfoertner opened 4 years ago

marvinpfoertner commented 4 years ago

Currently, (advanced) indexing, slicing, and masking of a Normal random variable with a LinearOperator mean and/or cov simply calls .todense() on the parameters and performs all operations on the matrices.

At least for simple indexing and slicing this can be sped up drastically by using broadcasted matrix vector products. We should also investigate whether efficient advanced indexing and masking is also possible.

marvinpfoertner commented 4 years ago

FYI @JonathanWenger