我想知道如何通过tf2.0中的tf.py_function计算高阶梯度。下面的示例(从tensorflow doc中略作修改)生成正确的dy_dx,而aa_x为None。谢谢。import tensorflow as tf if tf.abs(x) <= delta: else:
x = tf.constan
我试图在Tensorflow2.0中使用gradient_override_map。这里有一个,我也将用它作为示例。在2.0中,GradientTape可用于计算梯度,如下所示:print(tf.version.VERSION) # 2.0.0-alpha0
with tf.GradientTape() as tape:print(tape.gr