maziarraissi / PINNs

Physics Informed Deep Learning: Data-driven Solutions and Discovery of Nonlinear Partial Differential Equations
https://maziarraissi.github.io/PINNs
MIT License
3.98k stars 1.29k forks source link

can pytorch replace tensorflow #39

Open turthyan opened 3 years ago

turthyan commented 3 years ago

Does the program need to change much if it is to be developed on PyTorch?

Yogiraj587 commented 2 years ago

@turthyan Yess.... we have to change the code if it is to be developed in PyTorch. We have to import the pytorch library as torch and then we have to import appropriate modules from torch.