提交 05007f20 authored 作者: David Warde-Farley's avatar David Warde-Farley

Fixed missing line in Cholesky gradient procedure.

上级 b3cc2081
...@@ -404,6 +404,7 @@ class CholeskyGrad(Op): ...@@ -404,6 +404,7 @@ class CholeskyGrad(Op):
F[j, k] -= F[i, j] * L[i, k] F[j, k] -= F[i, j] * L[i, k]
for j in xrange(k + 1, N): for j in xrange(k + 1, N):
F[j, k] /= L[k, k] F[j, k] /= L[k, k]
F[k, k] -= L[j, k] * F[j, k]
F[k, k] /= (2 * L[k, k]) F[k, k] /= (2 * L[k, k])
else: else:
raise NotImplementedError("gradient only implemented for " raise NotImplementedError("gradient only implemented for "
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论