提交 c255a119 authored 作者: bergstra@tikuanyin's avatar bergstra@tikuanyin

fixed the failing reshape test

上级 eb6569b5
...@@ -1714,7 +1714,9 @@ def test_reshape(): ...@@ -1714,7 +1714,9 @@ def test_reshape():
assert numpy.all(a_val == a_val_copy) assert numpy.all(a_val == a_val_copy)
# verify gradient # verify gradient
utt.verify_grad(Reshape(2), [a_val,numpy.asarray([2,3], dtype='float64')]) def just_vals(v):
return Reshape(2)(v, numpy.asarray([2,3], dtype='float64'))
utt.verify_grad(just_vals, [a_val])
def test_flatten_outdimNone(): def test_flatten_outdimNone():
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论