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

[lz1iang]

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

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

[chenglu (luke)]

不懂，问问：@snowkylin @huan

[snowkylin (雪麒)]

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

[lz1iang]

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