当前位置:  开发笔记 > 编程语言 > 正文

如何判断TF OP是否具有渐变?

如何解决《如何判断TFOP是否具有渐变?》经验,为你挑选了1个好方法。

我对在SparseTensortensorflow 中使用感兴趣,但是,我经常得到

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

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



1> Maxim..:

中有一个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

推荐阅读
ifx0448363
这个屌丝很懒,什么也没留下!
DevBox开发工具箱 | 专业的在线开发工具网站    京公网安备 11010802040832号  |  京ICP备19059560号-6
Copyright © 1998 - 2020 DevBox.CN. All Rights Reserved devBox.cn 开发工具箱 版权所有