我正在研究一个新的优化器,我设法完成了大部分过程。我唯一坚持的是找到gen_training_ops
。
显然这个文件是至关重要的,因为在Gradient Descent和Adagrad优化器的两个实现中,它们都使用从gen_training_ops
(training_ops.py
文件夹中的python/training
)的包装文件导入的函数。我无法在任何地方找到这个文件,所以我想我不明白并且在错误的地方搜索。我在哪里可以找到它? (或者特别是apply_adagrad
和apply_gradient_descent
的实现)
非常感谢 :)
从源代码构建TensorFlow时将生成该文件。它在此文件中标记为out
,
https://github.com/tensorflow/tensorflow/blob/master/tensorflow/python/BUILD#L912
如果你找到它,你会意识到它只是跳转到python / framework,其中实际的更新只是一个赋值操作然后被分组