Bug fix: PDEOperatorCartesianProd (#1925)

lululxvi · Jan 3, 2025 · a3c677c · a3c677c
1 parent b313d96
commit a3c677c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deepxde/data/pde_operator.py b/deepxde/data/pde_operator.py
@@ -307,7 +307,7 @@ def forward_call(trunk_input):
 
         losses_bc = zip(*losses_bc)
         losses_bc = [bkd.reduce_mean(bkd.stack(loss, 0)) for loss in losses_bc]
-        losses.append(losses_bc)
+        losses.extend(losses_bc)
         return losses
 
     def losses_train(self, targets, outputs, loss_fn, inputs, model, aux=None):