出现致命错误 "JVP does not exist. Differential-first differentiation APIs are experimental and should not be used."

Question

我在 Xcode 中安装了 TensorFlow 工具链，并尝试 运行 从 S4TF GitHub 页面中获取示例片段。我收到以下错误：

JVP does not exist. Differential-first differentiation APIs are experimental and should not be used.

然后我尝试在 Google Colab 上做同样的事情，但得到了同样的错误。这是我尝试 运行:

的代码片段

@differentiable
func f(_ x: Float) -> Float {
    x * x
}
let dfdx = derivative(of: f)
dfdx(3) // 6

我知道这是我第一次 运行ning S4TF 并且我不知道很多事情，所以这可能是我的一个愚蠢的错误:)

Answer 1

您看到该错误是因为您正在尝试采用正向模式导数，并且 Swift 自动微分系统尚未实现对此的支持。当前实现支持反向模式微分，因此您可以使用 gradient 运算符来获取您想要的结果：

@differentiable
func f(_ x: Float) -> Float {
    x * x
}
let dfdx = gradient(of: f)
dfdx(3) // 6

正向模式差异化支持正在进行中，但目前的重点是逆向模式实现的上游并首先对其进行完善。作为免责声明，我在 Swift 为 TensorFlow 团队工作。