Use ravel instead of flatten for vec()

7 years ago · 7342c8efd7
parent d3c22006bd
commit 7342c8efd7
1 changed files with 1 additions and 1 deletions
--- a/fdfd_tools/vectorization.py
+++ b/fdfd_tools/vectorization.py
@ -27,7 +27,7 @@ def vec(f: field_t) -> vfield_t:
    """
    if numpy.any(numpy.equal(f, None)):
        return None
-    return numpy.hstack(tuple((fi.flatten(order='C') for fi in f)))
+    return numpy.hstack(tuple((fi.ravel(order='C') for fi in f)))


 def unvec(v: vfield_t, shape: numpy.ndarray) -> field_t: