如何实现PINN的tensorflow2.X版本,主要是对整个神经网络进行求导


如图是PINN1.0版本的代码,net_u是神经网络,net_f是通过对整个net_u进行求导构造的类似微分方程的网络。
请问tensorflow2.X版本可以实现类似的网络吗?
第二个就是自定义的损失函数,

这里分别对应net_u、net_f的误差,这部分我好像看到有tf2.x版本可以实现。
pinn的源代码连接:https://github.com/maziarraissi/PINNs,是tf1.x版本的~

1 Like

不懂,问问:@snowkylin @huan

我没有用过这个模型,不过高阶求导肯定是可以的。参考 https://mp.weixin.qq.com/s/C-f2vB2hd6At1h71GmCHQw

1 Like

大佬的回复真是精准啊,直击要害,谢谢啊

1 Like