add comment

lululxvi · Dec 17, 2024 · e7f2bb1 · e7f2bb1
1 parent 4d4106c
commit e7f2bb1
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/deepxde/data/pde_operator.py b/deepxde/data/pde_operator.py
@@ -267,6 +267,7 @@ def forward_call(trunk_input):
                 return aux[0]((inputs[0], trunk_input))
 
             if self.pde.pde is not None:
+                # f has the shape (`N1`, `N2`), inputs[1] has the shape (`N2`, `dim2`), outputs has the shape (`N1`, `N2`)
                 f = self.pde.pde(
                     inputs[1], (outputs, forward_call), model.net.auxiliary_vars
                 )