From 95818ccfa5f712a72b652cd0a369909b29bd3da3 Mon Sep 17 00:00:00 2001 From: Anran Jiao <38322669+anranjiao@users.noreply.github.com> Date: Sun, 1 Sep 2024 11:14:22 -0400 Subject: [PATCH] Fix a bug in quadruple.py --- deepxde/data/quadruple.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/deepxde/data/quadruple.py b/deepxde/data/quadruple.py index 2e9fc0ee4..efbe91767 100644 --- 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