A small update

lululxvi · Dec 18, 2023 · 661709a · 661709a
1 parent c34e2b7
commit 661709a
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/examples/pinn_forward/diffusion_1d_resample.py b/examples/pinn_forward/diffusion_1d_resample.py
@@ -32,19 +32,19 @@ def pde(x, y):
     #     + torch.exp(-x[:, 1:])
     #     * (torch.sin(np.pi * x[:, 0:1]) - np.pi ** 2 * torch.sin(np.pi * x[:, 0:1]))
     # )
-    # Backend paddle
+    # Backend jax
     # return (
     #     dy_t
     #     - dy_xx
-    #     + paddle.exp(-x[:, 1:])
-    #     * (paddle.sin(np.pi * x[:, 0:1]) - np.pi ** 2 * paddle.sin(np.pi * x[:, 0:1]))
+    #     + jnp.exp(-x[:, 1:])
+    #     * (jnp.sin(np.pi * x[..., 0:1]) - np.pi ** 2 * jnp.sin(np.pi * x[..., 0:1]))
     # )
-    # Backend jax
+    # Backend paddle
     # return (
     #     dy_t
     #     - dy_xx
-    #     + jnp.exp(-x[:, 1:])
-    #     * (jnp.sin(np.pi * x[..., 0:1]) - np.pi ** 2 * jnp.sin(np.pi * x[..., 0:1]))
+    #     + paddle.exp(-x[:, 1:])
+    #     * (paddle.sin(np.pi * x[:, 0:1]) - np.pi ** 2 * paddle.sin(np.pi * x[:, 0:1]))
     # )