f-dangel / curvlinops / 8394390320

90%

push

github

[ADD] `trace`, `det`, `logdet`, and `frobenius_norm` property to `KFACLinearOperator` (#95)

* Add tests for torch_matmat and torch_matvec

* Add torch_matmat and torch_matvec methods to KFACLinearOperator

* Nicer if-statement

Co-authored-by: sourcery-ai[bot] <58596630+sourcery-ai[bot]@users.noreply.github.com>

* Fix doc string

* Add tests for torch_matmat and torch_matvec for KFACInverseLinearOperator

* Add torch_matmat and torch_matvec to KFACInverseLinearOperator

* Increase test tolerance

* Deduce torch_matmat return type from input type

* Remove device check

* Make constant explicit

* Remove leftover import

* Add support for list inputs to torch_matvec

* Add tests for list inputs to torch_matmat and torch_matvec

* Address review feedback

* Switch to case for kfac tests and minor fixes

* Fix flake8

* Add test of KFAC trace property

* Add KFAC trace property

* Convert docstring to raw string

* Add KFAC frobenius_norm property test

* Add KFAC frobenius_norm property

* Add KFAC det property test

* Add KFAC det property and small refactor

* Fix det computation

* Fix KFAC det test and add logdet test

* Add KFAC logdet property

* Address review comments

* Switch from len to shape[0]

* Fix typo

---------

Co-authored-by: sourcery-ai[bot] <58596630+sourcery-ai[bot]@users.noreply.github.com>

68 of 74 new or added lines in 1 file covered. (91.89%)

1084 of 1211 relevant lines covered (89.51%)

0.9 hits per line