提交 db7c8b58 authored 作者: Frederic's avatar Frederic 提交者: Marc-Alexandre Cote

pep8

上级 1e141e5c
......@@ -50,7 +50,7 @@ class TestGpuCumsum(theano.tensor.tests.test_extra_ops.TestCumsumOp):
f = theano.function([x], cumsum(x[::2]), mode=self.mode)
a = np.random.randint(10, size=(42,)).astype("float32")
assert np.allclose(np.cumsum(a[::2]), f(a))
# Negative strides
f = theano.function([x], cumsum(x[::-1]), mode=self.mode)
a = np.random.randint(10, size=(42,)).astype("float32")
......
......@@ -38,7 +38,6 @@ class TestCumsumOp(utt.InferShapeTester):
f = theano.function([x], cumsum(x, axis=axis))
assert np.allclose(np.cumsum(a, axis=axis), f(a))
def test_infer_shape(self):
x = T.tensor3('x')
a = np.random.random((3, 5, 2)).astype(config.floatX)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论