Tensorflow的梯度覆盖映射函数

2024-06-16 08:55:14 发布

您现在位置:Python中文网/ 问答频道 /正文

有人能解释一下TensorFlow中的gradient_override_map函数吗? 我不能准确地理解它的用法。

我认为代码用法如下:

with G.gradient_override_map({"Floor": "Identity"}):
    return tf.reduce_mean(SomeVals) * SomeOtherVal

这里到底发生了什么?什么是Identity


Tags: 函数代码mapreduce用法returntftensorflow