jan/fdfd_tools
2
2
Fork 1
Code Issues Pull requests Releases Wiki Activity

Use ravel instead of flatten for vec()

Browse source
This commit is contained in:
jan 2017-09-24 19:13:10 -07:00
commit 7342c8efd7
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
fdfd_tools/vectorization.py
View file

@ -27,7 +27,7 @@ def vec(f: field_t) -> vfield_t:
""" """
if numpy.any(numpy.equal(f, None)): if numpy.any(numpy.equal(f, None)):
return 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: def unvec(v: vfield_t, shape: numpy.ndarray) -> field_t: