我可以在张量流中应用计算梯度吗?

时间:2016-11-10 17:54:37

标签: python machine-learning tensorflow deep-learning

我想要做的是在不同的机器上模拟反向传播过程,从一台机器,我从第3层d(layer3_output)/d(layer2_output)得到渐变作为一个numpy数组,我怎么能得到{{1}有效地给出我收到的渐变并传递给前一层?

1 个答案:

答案 0 :(得分:1)

当您创建网络并附加一些损失时,请在优化程序上调用最小化,优化程序(在引擎盖下)调用“apply_gradients”。此函数将渐变计算操作添加到图形中。您所要做的就是请求负责您的偏导数的操作,并通过feed_dict选项传递预先计算的偏导数。使用张量板可视化您的图形并调查您感兴趣的渐变名称。默认情况下,它们将位于“渐变”名称范围内,并且每个操作的命名将类似于您的操作,因此{{1}的行之类的内容等等。