首页 文章

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

提问于
浏览
0

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

1 回答

  • 1

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

相关问题