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

Test grad wrt eigenvalues and eigenvectors separately.

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