我有兴趣在 tensorflow 中使用 SparseTensor
,但是,我经常得到 p>
LookupError: No gradient defined for operation ...
显然,对于稀疏张量的许多操作都没有定义梯度计算。在实际编写和运行我的代码之前,是否有任何简单的方法来检查操作是否具有梯度?
最佳答案
在tensorflow.python.framework.ops
中有一个get_gradient_function
函数。它接受一个操作并返回相应的梯度操作。示例:
import tensorflow as tf
from tensorflow.python.framework.ops import get_gradient_function
a = tf.add(1, 2, name="Add_these_numbers")
b = tf.multiply(a, 3, name='mult')
mult = tf.get_default_graph().get_operation_by_name('mult')
print(get_gradient_function(mult)) # <function _MulGrad at 0x7fa29950dc80>
tf.stop_gradient(a, name='stop')
stop = tf.get_default_graph().get_operation_by_name('stop')
print(get_gradient_function(stop)) # None
关于python - 我如何判断 tf op 是否具有梯度?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/48418029/