提交 3768e76b authored 作者: Frederic's avatar Frederic

re-added yiel as per review.

上级 735263e4
...@@ -82,11 +82,13 @@ def test_cholesky_grad(): ...@@ -82,11 +82,13 @@ def test_cholesky_grad():
if config.floatX == "float64": if config.floatX == "float64":
eps = 2e-8 eps = 2e-8
# Check the default. # Check the default.
utt.verify_grad(cholesky, [pd], 3, rng, eps=eps) yield (lambda: utt.verify_grad(cholesky, [pd], 3, rng, eps=eps))
# Explicit lower-triangular. # Explicit lower-triangular.
utt.verify_grad(Cholesky(lower=True), [pd], 3, rng, eps=eps) yield (lambda: utt.verify_grad(Cholesky(lower=True), [pd], 3,
rng, eps=eps))
# Explicit upper-triangular. # Explicit upper-triangular.
utt.verify_grad(Cholesky(lower=False), [pd], 3, rng, eps=eps) yield (lambda: utt.verify_grad(Cholesky(lower=False), [pd], 3,
rng, eps=eps))
def test_cholesky_and_cholesky_grad_shape(): def test_cholesky_and_cholesky_grad_shape():
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论