linear_operator.old

linear_operator.old copied to clipboard

Return (inv) root of KroneckerProductAddedDiagLinearOperator as lazy

1

This is a clone of https://github.com/cornellius-gp/gpytorch/pull/1430 Previously, `_root_decomposition` and `_inv_root_decomposition` were returning the (inv) root from the eigendecomposition as a dense tensor, which can be inefficient. This now returns the...

Balandat

Logging https://github.com/cornellius-gp/gpytorch/issues/1395 here as well. It's probably more useful for LinearOperator.

wjmaddox

Clone of gpytorch PR: https://github.com/cornellius-gp/gpytorch/pull/1394 so that its not forgotten. I can add it in for PSDLinearOperators later.

wjmaddox

gpleiss

gpleiss