Fix a bug in quadruple.py

lululxvi · Sep 1, 2024 · 95818cc · 95818cc
1 parent 7f1b87f
commit 95818cc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deepxde/data/quadruple.py b/deepxde/data/quadruple.py
@@ -94,7 +94,7 @@ def train_next_batch(self, batch_size=None):
  self.train_x[0][indices_branch],
  self.train_x[1][indices_branch],
  self.train_x[2][indices_trunk],
- ), self.train_y[indices_branch, indices_trunk]
+ ), self.train_y[indices_branch][:, indices_trunk]
 
  def test(self):
  return self.test_x, self.test_y