十进制可微操作列表

Question

是否有可微的Tensorflow操作的主列表（即，将自动区分）？在

还有两种表达方式：

• 未设置ops.NoGradient的操作列表。在
• 不会触发LookupError的操作列表。在

例如，我假设所有的控制流操作都是不可微的（例如，tf.where）。除了手动运行所有这些程序来查看它们是否抛出LookupError之外，我如何才能找到这一点呢。在

“常识”不是一个有效的答案。在

谢谢。在

编辑：

tf.where是可微的，所以我的直觉是错误的。也许这里正确的问题是张量流中哪些ops是不可微的。在

谢谢。在

Answer 1

我已经使用python代码设计了可微和不可微操作的完整列表。在

你可以在这里找到紧凑的列表。还有生成它的代码。在

https://github.com/Mainak431/List-of-Differentiable OPs-and-Non-differentiable-OPs in-Tensorflow

Answer 2

不，没有列表（您可以是第一个创建它的人）。另外，据我所知，每个函数的文档也没有告诉它（^{}是不可微的，但没有说明它）。在

除了您建议的方法之外，您还可以从源代码中提取这些数据。例如，所有实现了渐变的操作都在方法声明前面有^{}。{ops没有梯度

不相关，但probably helpful。在