如何实现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]

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

[Sergio.6X]

请问大佬这部分可以用tf2实现么，最近导师让我用tf2复现源码中PINN的训练。试了自定义损失函数、自定义模型还是没办法复现

[inn]

同学，你最后实现了吗，我导师最近也让我做这个

[inn]

请问怎么实现的，我最近也在做这个

[inn]

！兄弟，快教教我，但我高阶求导完是Nonetype

[inn]

(post deleted by author)

[inn]

现在问题是一阶求导可以了，但二阶始终是Nonetype，求大佬指点一下

[inn]

加了，兄弟！