提交 4a2ad345 authored 作者: David Warde-Farley's avatar David Warde-Farley

Added a test for CholeskyGrad.

上级 e382e04d
...@@ -63,6 +63,18 @@ def test_cholesky(): ...@@ -63,6 +63,18 @@ def test_cholesky():
yield check_upper_triangular, pd, ch_f yield check_upper_triangular, pd, ch_f
def test_cholesky_grad():
rng = numpy.random.RandomState(utt.fetch_seed())
r = rng.randn(5, 5)
pd = numpy.dot(r, r.T)
# Check the default.
utt.verify_grad(cholesky, [pd], rng=rng)
# Explicit lower-triangular.
utt.verify_grad(Cholesky(lower=True), [pd], rng=rng)
# Explicit upper-triangular.
utt.verify_grad(Cholesky(lower=False), [pd], rng=rng)
def test_inverse_correctness(): def test_inverse_correctness():
rng = numpy.random.RandomState(utt.fetch_seed()) rng = numpy.random.RandomState(utt.fetch_seed())
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论