提交 60071f34 authored 作者: Matt Graham's avatar Matt Graham

Adding reference to docstring.

上级 738d9e86
...@@ -62,6 +62,16 @@ class Cholesky(Op): ...@@ -62,6 +62,16 @@ class Cholesky(Op):
z[0] = scipy.linalg.cholesky(x, lower=self.lower).astype(x.dtype) z[0] = scipy.linalg.cholesky(x, lower=self.lower).astype(x.dtype)
def grad(self, inputs, gradients): def grad(self, inputs, gradients):
"""
Cholesky decomposition reverse-mode gradient update.
Notes
-----
Symbolic expression for reverse-mode Cholesky gradient taken from [1]_
[1] I. Murray, "Differentiation of the Cholesky decomposition",
http://arxiv.org/abs/1602.07527
"""
x = inputs[0] x = inputs[0]
dz = gradients[0] dz = gradients[0]
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论