From f08f21d9dfb18466fe5e6428ac977b85e860795d Mon Sep 17 00:00:00 2001 From: Zongren Zou <31862613+ZongrenZou@users.noreply.github.com> Date: Mon, 18 Dec 2023 17:30:38 -0500 Subject: [PATCH] A small update --- examples/pinn_forward/diffusion_1d_exactBC.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/pinn_forward/diffusion_1d_exactBC.py b/examples/pinn_forward/diffusion_1d_exactBC.py index 2df7dd985..014f45057 100644 --- a/examples/pinn_forward/diffusion_1d_exactBC.py +++ b/examples/pinn_forward/diffusion_1d_exactBC.py @@ -67,10 +67,10 @@ def func(x): lambda x, y: x[:, 1:2] * (1 - x[:, 0:1] ** 2) * y + tf.sin(np.pi * x[:, 0:1]) # Backend pytorch # lambda x, y: x[:, 1:2] * (1 - x[:, 0:1] ** 2) * y + torch.sin(np.pi * x[:, 0:1]) - # Backend paddle - # lambda x, y: x[:, 1:2] * (1 - x[:, 0:1] ** 2) * y + paddle.sin(np.pi * x[:, 0:1]) # Backend jax # lambda x, y: x[..., 1:2] * (1 - x[..., 0:1] ** 2) * y + jnp.sin(np.pi * x[..., 0:1]) + # Backend paddle + # lambda x, y: x[:, 1:2] * (1 - x[:, 0:1] ** 2) * y + paddle.sin(np.pi * x[:, 0:1]) ) model = dde.Model(data, net)