提交 a2f0fb30 authored 作者: Razvan Pascanu's avatar Razvan Pascanu

add tests to check the gradient of set_subtensor

上级 c7f113b4
...@@ -115,3 +115,32 @@ class Test_inc_subtensor(unittest.TestCase): ...@@ -115,3 +115,32 @@ class Test_inc_subtensor(unittest.TestCase):
inc_slice(2, 1), inc_slice(2, 1),
(numpy.asarray([[0,1],[2,3],[4,5.]]), (numpy.asarray([[0,1],[2,3],[4,5.]]),
numpy.asarray(9.),)) numpy.asarray(9.),))
def test_grad_set(self):
a = T.dvector()
b = T.dvector()
def set_slice(*s):
def just_numeric_args(a,b):
return T.set_subtensor(a[s], b)
return just_numeric_args
# vector
utt.verify_grad(
set_slice(slice(2,4,None)),
(numpy.asarray([0,1,2,3,4,5.]),
numpy.asarray([9,9.]),))
# matrix
utt.verify_grad(
set_slice(slice(1,2,None), slice(None, None, None)),
(numpy.asarray([[0,1],[2,3],[4,5.]]),
numpy.asarray([[9,9.]]),))
#single element
utt.verify_grad(
set_slice(2, 1),
(numpy.asarray([[0,1],[2,3],[4,5.]]),
numpy.asarray(9.),))
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论