提交 4f349e93 authored 作者: abalkin's avatar abalkin

Added TensorVariable.ravel().

上级 68c4085f
...@@ -1634,6 +1634,9 @@ class _tensor_py_operators: ...@@ -1634,6 +1634,9 @@ class _tensor_py_operators:
def flatten(self, ndim=1): def flatten(self, ndim=1):
return flatten(self, ndim) return flatten(self, ndim)
def ravel(self):
return flatten(self)
# CASTING # CASTING
def astype(self, dtype): def astype(self, dtype):
return cast(self, dtype) return cast(self, dtype)
......
...@@ -7066,6 +7066,11 @@ class TestTensorInstanceMethods(unittest.TestCase): ...@@ -7066,6 +7066,11 @@ class TestTensorInstanceMethods(unittest.TestCase):
x, _ = self.vals x, _ = self.vals
self.assertTrue(numpy.all(X.trace().eval({X: x}) == x.trace())) self.assertTrue(numpy.all(X.trace().eval({X: x}) == x.trace()))
def test_ravel(self):
X, _ = self.vars
x, _ = self.vals
self.assertTrue(numpy.all(X.ravel().eval({X: x}) == x.ravel()))
if __name__ == '__main__': if __name__ == '__main__':
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论