提交 fbb90f59 authored 作者: Jeremiah Lowin's avatar Jeremiah Lowin

remove tensordot_grad test (no longer needed)

上级 51b5463f
......@@ -5589,25 +5589,6 @@ class test_tensordot(unittest.TestCase):
f3(aval, bval)))
utt.verify_grad(self.TensorDot(axes), [aval, bval])
def test_tensordot_grad(self):
# We test it manually as we recreate the op in the make_node
amat = matrix()
bmat = matrix()
gzmat = matrix()
axes = 1
aval = rand(4, 5)
bval = rand(5, 3)
gzval = rand(4, 3)
f1 = inplace_func([amat, bmat, gzmat], tensordot_grad(axes)(
amat, bmat, gzmat))
f2 = inplace_func([amat, bmat, gzmat], tensordot_grad(((1, ), (
0,)))(amat, bmat, gzmat))
o1 = f1(aval, bval, gzval)
o2 = f2(aval, bval, gzval)
self.assertTrue(numpy.allclose(o1[0], o2[0]))
self.assertTrue(numpy.allclose(o1[1], o2[1]))
def test_smallest_stack():
sx, sy = dscalar(), dscalar()
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论