【架构分析】Tensorflow 自动微分源码分析
文章 链接 对几种不同类型的微分做了很详细的图文说明,值得一看
Tensorflow 求解梯度(微分)的核心函数代码分析如下
@tf_export(v1=["gradients"])
def gradients(ys,
xs,
grad_ys=None,
name="gradients",
colocate_gradients_with_ops=False,
gate_gradients=False,
aggregation_method=None,
stop_gradients=None,
unconnected_gradients=UnconnectedGradients.NONE):
'''
Args:
ys: A `Tensor` or list of tensors to be differenti
共有 0 条评论