知乎话题
TensorFlow的求导,实际上是先提供每一个op求导的数学实现(人手工写的函数代码),然后使用链式法则自动求出整个表达式的导数。
op的forward和backward,都是用矩阵做输入和输出的,便于GPU调库加速。
很好的示例代码:
开源计算图示例(1);开源计算图示例(2);开源计算图示例(3)
四种求导方式的介绍;论文:Automatic Differentiation in Machine Learning: a Survey
TensorFlow的求导,实际上是先提供每一个op求导的数学实现(人手工写的函数代码),然后使用链式法则自动求出整个表达式的导数。
op的forward和backward,都是用矩阵做输入和输出的,便于GPU调库加速。
很好的示例代码:
开源计算图示例(1);开源计算图示例(2);开源计算图示例(3)
四种求导方式的介绍;论文:Automatic Differentiation in Machine Learning: a Survey