From 62726e1ec3cc7503de62bbbb33ed9db675fc64db Mon Sep 17 00:00:00 2001 From: Anton Khirnov Date: Wed, 24 May 2017 16:03:43 +0200 Subject: series_expansion: implement 2D evaluation --- series_expansion.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/series_expansion.py b/series_expansion.py index 76adc30..15ee79d 100644 --- a/series_expansion.py +++ b/series_expansion.py @@ -67,5 +67,10 @@ class SeriesExpansion(object): for c, val in zip(self._coeffs, basis_vals[0]): ret += val * c return ret + elif self._dim == 2: + for i in xrange(self._coeffs.shape[0]): + for j in xrange(self._coeffs.shape[1]): + ret += self._coeffs[i, j] * np.outer(basis_vals[0][i], basis_vals[1][j]) + return ret else: raise NotImplementedError('Unsupported number of dimensions') -- cgit v1.2.3