我对在SparseTensor
tensorflow 中使用感兴趣,但是,我经常得到
LookupError:没有为操作定义渐变...
显然,对于稀疏张量的许多运算没有定义梯度计算。在实际编写和运行我的代码之前,是否有任何简单的方法来检查操作是否具有渐变?
中有一个get_gradient_function
功能tensorflow.python.framework.ops
。它接受一个op并返回相应的渐变op。例:
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)) #
tf.stop_gradient(a, name='stop')
stop = tf.get_default_graph().get_operation_by_name('stop')
print(get_gradient_function(stop)) # None