From 82341db0ad767bc61780ae219710bf2c719795b7 Mon Sep 17 00:00:00 2001
From: rmojgani <rmojgani@users.noreply.github.com>
Date: Wed, 16 Feb 2022 16:07:01 -0600
Subject: [PATCH] bug in reshape of u

bug in reshape of u, important for the case where n!=m
---
 LinearAdvectionPINN.ipynb | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/LinearAdvectionPINN.ipynb b/LinearAdvectionPINN.ipynb
index 7d2ae00..582e0af 100644
--- a/LinearAdvectionPINN.ipynb
+++ b/LinearAdvectionPINN.ipynb
@@ -259,7 +259,7 @@
         "\r\n",
         "u = model([np.expand_dims(T.flatten(), axis=1), \r\n",
         "           np.expand_dims(X.flatten(), axis=1)])\r\n",
-        "u = np.reshape(u, (m, n))"
+        "u = np.reshape(u, (n, m))"
       ],
       "execution_count": 7,
       "outputs": []
@@ -347,4 +347,4 @@
       ]
     }
   ]
-}
\ No newline at end of file
+}