update code

lululxvi · Dec 28, 2024 · 101bc31 · 101bc31
1 parent e20364c
commit 101bc31
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/deepxde/data/pde_operator.py b/deepxde/data/pde_operator.py
@@ -306,8 +306,8 @@ def forward_call(trunk_input):
             losses_bc.append(losses_i)
 
         losses_bc = zip(*losses_bc)
-        for loss in losses_bc:
-            losses.append(bkd.reduce_mean(bkd.stack(loss, 0)))
+        losses_bc = [bkd.reduce_mean(bkd.stack(loss, 0)) for loss in losses_bc]
+        losses.extend(losses_bc)
         return losses
 
     def losses_train(self, targets, outputs, loss_fn, inputs, model, aux=None):