diff --git a/fdfd_tools/vectorization.py b/fdfd_tools/vectorization.py index 2377d39..1c2134a 100644 --- 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: