From e7f2bb16eb8fcce07331a03c6a1e5fbf38e76574 Mon Sep 17 00:00:00 2001 From: Jerry-Jzy Date: Mon, 16 Dec 2024 22:42:02 -0500 Subject: [PATCH] add comment --- deepxde/data/pde_operator.py | 1 + 1 file changed, 1 insertion(+) diff --git a/deepxde/data/pde_operator.py b/deepxde/data/pde_operator.py index 174df29eb..5956128a8 100644 --- 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 )