提交 143103d3 authored 作者: abalkin's avatar abalkin 提交者: Frederic

Test grad wrt eigenvalues and eigenvectors separately.

上级 4d6fa38b
......@@ -502,7 +502,5 @@ class test_Eigh(test_Eig):
op = eigh
def test_grad(self):
S = self.S
def fun(x):
w, v = self.op(x + x.T)
return w.sum() + v.sum()
utt.verify_grad(fun, [S], rng=self.rng)
utt.verify_grad(lambda x: self.op(x + x.T)[0], [S], rng=self.rng)
utt.verify_grad(lambda x: self.op(x + x.T)[1], [S], rng=self.rng)
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论