Tensorflow什么操作是可区分的，什么不是？

Question

在Tensorflow中，很难确定函数是否可微分。例如，tf.argmax不可区分。我想知道是否有任何文件来指明哪些操作是可区分的？

Answer 1

对于数学运算，将渐变记录在此文件中： tensorflow/tensorflow/python/ops/math_grad.py

例如tf.argmax的梯度：

@ops.RegisterGradient("ArgMax")
def _ArgMaxGrad(op, grad):
  del op, grad
  return [None, None]

其他操作的渐变也可以在同一文件夹中找到。