fix poission 1d (PaddlePaddle#188)

zlynna · Sep 5, 2022 · 633f20c · 633f20c
1 parent cb2b574
commit 633f20c
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/paddlescience/pde/pde_poisson.py b/paddlescience/pde/pde_poisson.py
@@ -40,7 +40,14 @@ class Poisson(PDE):
 
     def __init__(self, dim=2, rhs=0.0, weight=1.0):
 
-        if dim == 2:
+        if dim == 1:
+            # independent and dependent variable
+            x = sympy.Symbol('x')
+            u = sympy.Function('u')(x)
+            super(Poisson, self).__init__([x], [u], weight)
+            self.add_equation(u.diff(x).diff(x), rhs)
+
+        elif dim == 2:
             # independent and dependent variable
             x = sympy.Symbol('x')
             y = sympy.Symbol('y')