编程语言
首页 > 编程语言> > python – 如何判断tf op是否有渐变?

python – 如何判断tf op是否有渐变?

作者:互联网

我有兴趣在tensorflow中使用SparseTensor,但是,我经常得到

LookupError: No gradient defined for operation …

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

解决方法:

tensorflow.python.framework.ops中有一个get_gradient_function函数.它接受一个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

标签:python,tensorflow,autodiff
来源: https://codeday.me/bug/20191006/1862617.html