Skip to content
Open
Changes from 2 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 5 additions & 1 deletion examples/pinn_forward/Poisson_Dirichlet_1d_exactBC.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
"""Backend supported: tensorflow.compat.v1, tensorflow, jax, paddle"""
"""Backend supported: tensorflow.compat.v1, tensorflow, jax, pytorch, paddle"""
import deepxde as dde
import numpy as np

Expand All @@ -18,6 +18,10 @@
import paddle

sin = paddle.sin
elif dde.backend.backend_name == "pytorch":
import torch

sin = torch.sin


def pde(x, y):
Expand Down