List of Differentiable Ops in Tensorflow

后端 未结 2 1075
情歌与酒
情歌与酒 2020-12-31 07:07

Is there a master list of Tensorflow ops that are differentiable (i.e., will auto-differentiate)?

Two other ways to phrase this:

  • List of ops that do no
相关标签:
2条回答
  • 2020-12-31 07:44

    No, there is no list (you can be the first one to create it). Also as far as I am aware, documentation of each function also does not tell it (tf.size is non-differentiable but does not tell about it).

    Apart from the way you suggested, you can also extract this data from the source code. For example all the ops that have gradient implemented, have @ops.RegisterGradient in front of the method declaration. For ops which do not have gradient you will have ops.NotDifferentiable(

    Not related, but probably helpful.

    0 讨论(0)
  • 2020-12-31 07:47

    I have devised the entire list of Differentiable and Non-Differentiable Ops using python code.

    You will find the compact list here. Also the code which generated it.

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

    0 讨论(0)
提交回复
热议问题