提交 f7028bde authored 作者: Roy Xue's avatar Roy Xue

add cumsum and cumprod test in theano.tensor.test.test_basic.py

上级 fe4f52c9
...@@ -6736,6 +6736,17 @@ class TestTensorInstanceMethods(unittest.TestCase): ...@@ -6736,6 +6736,17 @@ class TestTensorInstanceMethods(unittest.TestCase):
# Test equivalent advanced indexing # Test equivalent advanced indexing
assert_array_equal(X[:,indices].eval({X: x}), x[:,indices]) assert_array_equal(X[:,indices].eval({X: x}), x[:,indices])
def test_cumsum(self):
X, _ = self.vars
x, _ = self.vals
assert_array_equal(X.cumsum().eval({X: x}), x.cumsum())
def test_cumprod(self):
X, _ = self.vars
x, _ = self.vals
assert_array_equal(X.cumprod().eval({X: x}), x.cumprod())
def test_norm(): def test_norm():
x = theano.tensor.vector('x') x = theano.tensor.vector('x')
n = x.norm(2) n = x.norm(2)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论