TensorFlow首次快速体验

<dependency>
    <groupId>org.tensorflow</groupId>
    <artifactId>tensorflow</artifactId>
    <version>1.13.1</version>
</dependency>

f(x, y) = z = a*x + b*y

Graph graph = new Graph()

Operation a = graph.opBuilder("Const", "a")
        .setAttr("dtype", DataType.fromClass(Double.class))
        .setAttr("value", Tensor.<Double>create(3.0, Double.class))
        .build();
Operation b = graph.opBuilder("Const", "b")
        .setAttr("dtype", DataType.fromClass(Double.class))
        .setAttr("value", Tensor.<Double>create(2.0, Double.class))
        .build()

Operation x = graph.opBuilder("Placeholder", "x")
        .setAttr("dtype", DataType.fromClass(Double.class))
        .build();
Operation y = graph.opBuilder("Placeholder", "y")
        .setAttr("dtype", DataType.fromClass(Double.class))
        .build();

Operation ax = graph.opBuilder("Mul", "ax")
        .addInput(a.output(0))
        .addInput(x.output(0))
        .build();
Operation by = graph.opBuilder("Mul", "by")
        .addInput(b.output(0))
        .addInput(y.output(0))
        .build();
Operation z = graph.opBuilder("Add", "z")
        .addInput(ax.output(0))
        .addInput(by.output(0))
        .build();

Session session = new Session(graph);
Tensor<Double> tensor = session.runner().fetch("z")
        .feed("x", Tensor.create(3.0, Double.class))
        .feed("y", Tensor.create(6.0, Double.class))
        .run().get(0).expect(Double.class);
System.out.println(tensor.doubleValue());

Path path = Paths.get("tensor.model");
byte[] bytes = graph.toGraphDef();
Files.write(path, bytes);

Graph graph = new Graph();
byte[] bytes = Files.readAllBytes(path);
graph.importGraphDef(bytes);