是否可以使用 tf.GradientTape() 创建一个可以产生二阶、三阶或 n 阶梯度的函数？

问题描述

我一直在尝试实现一个可以使用 GradientTape 输出函数的 n 阶微分的函数。例如：

如果函数是 y = x^2 * sin(x) 且 x=7 且 n=8。

那么一阶微分是dy/dx，二阶微分是d2y/dx2。该函数将返回 dny/dnx。

我尝试在下面实现一些东西，但没有成功，因为 tf.GradientTape() 需要跟踪需要微分的方程。

def function(x):
  return (x**2) * tf.math.sin(x)

def gradient_by_order(function,x,n):
  
  with tf.GradientTape() as tape:
    y = function(x)
  
  d = tape.gradient(y,x)
 
  with tf.GradientTape() as tape:
    for i in range(n):
      d = tape.gradient(d,x)

x = tf.Variable(initial_value=5.)
gradient_by_order(function,10)

有没有办法使用递归或循环或任何其他方法来创建工作函数？ 非常感谢:)

编辑：我知道您可以通过嵌套 with tf.GradientTape() as tape 来获得 n 阶导数，就像下面 Susmit Agrawal 的回答中所示。我只是想知道是否有一种方法可以编写一个可以输出 n 阶导数的函数（假设存在这样的导数），因为我找不到。

解决方法

暂无找到可以解决该程序问题的有效方法，小编努力寻找整理中！

如果你已经找到好的解决方法，欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@）