gpt4 book ai didi

java - 使用Java中的OperationBuilder类在tensorflow-java中构建图

转载 作者:行者123 更新时间:2023-12-02 11:25:34 25 4
gpt4 key购买 nike

嗨,我正在关注 https://github.com/tensorflow/tensorflow/blob/master/tensorflow/java/src/main/java/org/tensorflow/examples/LabelImage.java 的官方 tensorflow 示例在java中构建我的计算图,而不是使用在python中构建的计算图。我看到他们构建了一个图来预处理图像,不需要任何反向传播。所以我的问题是,如果我想构建一个用于完整训练的计算图,我是否需要遵循相同的步骤(为操作创建静态 API),或者我是否需要做更多的事情来管理反向传播的梯度?

最佳答案

TensorFlow Java 绑定(bind)不支持自动微分。您可以通过手动添加适当的操作来自己计算梯度,但是Python中没有像tf.gradients()这样的魔法可以为您添加它们。

关于java - 使用Java中的OperationBuilder类在tensorflow-java中构建图,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/49657561/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com