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]

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