java tensorflow

// 导入 TensorFlow 库
import org.tensorflow.Tensor;
import org.tensorflow.Session;
import org.tensorflow.Graph;

public class JavaTensorFlowExample {
    public static void main(String[] args) {
        // 创建一个 TensorFlow 图形对象
        try (Graph graph = new Graph()) {
            // 定义计算图中的操作
            graph.opBuilder("Const", "ConstInput")
                .setAttr("dtype", Tensor.create(1.0).dataType())
                .setAttr("value", Tensor.create(1.0))
                .build();

            graph.opBuilder("Const", "ConstWeight")
                .setAttr("dtype", Tensor.create(2.0).dataType())
                .setAttr("value", Tensor.create(2.0))
                .build();

            graph.opBuilder("Add", "Add")
                .addInput(graph.operation("ConstInput").output(0))
                .addInput(graph.operation("ConstWeight").output(0))
                .build();

            // 创建一个会话来运行图形
            try (Session session = new Session(graph)) {
                // 运行图形并获取结果
                Tensor result = session.runner()
                    .fetch("Add")
                    .run().get(0);

                // 打印结果
                System.out.println("Result: " + result.data().asFloat());
            }
        }
    }
}

解释说明：

• 导入库：首先导入了 TensorFlow 的相关类，包括 Tensor, Session, 和 Graph。
• 创建图形：使用 Graph 类创建了一个新的计算图，并在其中定义了两个常量节点 (ConstInput 和 ConstWeight) 以及一个加法操作 (Add)。
• 创建会话：通过 Session 类创建了一个会话，用于执行计算图中的操作。
• 运行图形：使用 session.runner() 来指定要运行的操作，并通过 fetch("Add") 获取加法操作的结果。
• 打印结果：最后，将结果转换为浮点数并打印出来。

这段代码展示了如何在 Java 中使用 TensorFlow 进行简单的数值计算。