Tensorflow tf.GradientTape().gradient的自定义聚合?(TF2.0)

Tensorflow tf.GradientTape().gradient的自定义聚合?(TF2.0),tensorflow,tensorflow2.0,Tensorflow,Tensorflow2.0,据我所知,tf.gradients函数提供了选择聚合方法的选项,用于汇总来自多个源的梯度。 然而,根据Tensorflow API文档,tf.GradientTape().gradient方法没有这样的选项。 因此,我的问题如下: 有没有办法更改tf.GradientTape().gradient中的聚合方法 如果没有,是否有任何方法可以通过与即时执行兼容的自定义聚合方法获得梯度

据我所知,tf.gradients函数提供了选择聚合方法的选项,用于汇总来自多个源的梯度。

然而,根据Tensorflow API文档,tf.GradientTape().gradient方法没有这样的选项。

因此,我的问题如下:

  • 有没有办法更改tf.GradientTape().gradient中的聚合方法

  • 如果没有,是否有任何方法可以通过与即时执行兼容的自定义聚合方法获得梯度