tensorflow C 或 C++ API 中是否存在等效于 tf.gradients function 的值？

Question

我想在 C 或 C++ API 中实现一个 tensorflows function tf.gradients ？ Tensorflow C++ 拥有世界上最糟糕的文档，而 C API 则根本没有记录。 你能建议我是否有一个实现或者我应该使用哪个 API 部分来自己开发这个。

Answer 1

python function tf.gradients调用的 C 函数可以在以下 header 文件中找到：

#include "tensorflow/core/gradients/grad_ops.h"

您可以使用tensorflow::GradientTape来初始化一个磁带，它可以调用gradient function 来计算张量y相对于x的梯度。