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

Added TensorVariable.ravel().

上级 68c4085f
......@@ -1634,6 +1634,9 @@ class _tensor_py_operators:
def flatten(self, ndim=1):
return flatten(self, ndim)
def ravel(self):
return flatten(self)
# CASTING
def astype(self, dtype):
return cast(self, dtype)
......
......@@ -7066,6 +7066,11 @@ class TestTensorInstanceMethods(unittest.TestCase):
x, _ = self.vals
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__':
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论