summaryrefslogtreecommitdiff
path: root/numpy/polynomial/hermite_e.py
diff options
context:
space:
mode:
authorEric Wieser <wieser.eric@gmail.com>2019-08-24 14:29:42 -0700
committerEric Wieser <wieser.eric@gmail.com>2019-09-14 19:55:48 -0700
commite443198415038a5995a15c6c32b6de0919c95f70 (patch)
treed0c063724c5c7c872b725f305c37549b62be71f0 /numpy/polynomial/hermite_e.py
parent99c62dc44a70af407f266eb30b756b297c215899 (diff)
downloadnumpy-e443198415038a5995a15c6c32b6de0919c95f70.tar.gz
MAINT: polynomial: Add an N-d vander implementation used under the hood of the vander2d and vander3d functions
The generalization is not exposed in the public API yet, but it could be if the need arises. The shape / dtype conversion logic is left as is for now, even if it might be broken.
Diffstat (limited to 'numpy/polynomial/hermite_e.py')
-rw-r--r--numpy/polynomial/hermite_e.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/numpy/polynomial/hermite_e.py b/numpy/polynomial/hermite_e.py
index b1cc2d3ab..c5a0a05a2 100644
--- a/numpy/polynomial/hermite_e.py
+++ b/numpy/polynomial/hermite_e.py
@@ -1186,7 +1186,7 @@ def hermevander2d(x, y, deg):
.. versionadded:: 1.7.0
"""
- return pu._vander2d(hermevander, x, y, deg)
+ return pu._vander_nd_flat((hermevander, hermevander), (x, y), deg)
def hermevander3d(x, y, z, deg):
@@ -1240,7 +1240,7 @@ def hermevander3d(x, y, z, deg):
.. versionadded:: 1.7.0
"""
- return pu._vander3d(hermevander, x, y, z, deg)
+ return pu._vander_nd_flat((hermevander, hermevander, hermevander), (x, y, z), deg)
def hermefit(x, y, deg, rcond=None, full=False, w=None):