如何判断某个梯度是否有梯度？

Question

我有兴趣在张量流中使用header("HTTP/1.0 404 Not Found"); ，但是，我经常得到

  

LookupError：没有为操作定义渐变...

显然没有为稀疏张量的许多操作定义梯度计算。在实际编写和运行我的代码之前，有没有简单的方法来检查op是否具有渐变？

Answer 1

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))  # <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