如何在java中读取tensorflow模型的输出

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

import org.tensorflow.Graph;
import org.tensorflow.Session;
import org.tensorflow.Tensor;
import org.tensorflow.TensorFlow;

public class TensorFlowModelReader {
    public static void main(String[] args) {
        try (Graph graph = new Graph()) {
            byte[] graphBytes = Files.readAllBytes(Paths.get("model.pb"));
            graph.importGraphDef(graphBytes);

            try (Session session = new Session(graph)) {
                // 执行计算图
                Tensor<?> output = session.runner()
                        .feed("input_name", inputTensor) // 设置输入
                        .fetch("output_name") // 设置输出
                        .run()
                        .get(0);
                        
                // 处理输出
                // ...                
            }
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
}

float[] outputData = new float[output.numElements()];
output.copyTo(outputData);