shuren/fdfd_tools
1
0
Fork 0
forked from jan/fdfd_tools
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
parent d3c22006bd
commit 7342c8efd7
1 changed files with 1 additions and 1 deletions
Show Stats 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: