如何将tensorflow 2.x模型导入Java？

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

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

String modelPath = "/path/to/your/model";
Graph graph = new Graph();
byte[] graphBytes = Files.readAllBytes(Paths.get(modelPath));
graph.importGraphDef(graphBytes);
Session session = new Session(graph);

float[] inputData = {1.0f, 2.0f, 3.0f};
long[] inputShape = {1, 3}; // 根据模型的输入形状设置
Tensor inputTensor = Tensor.create(inputShape, FloatBuffer.wrap(inputData));

Tensor outputTensor = session.runner()
    .feed("input_tensor_name", inputTensor) // 根据模型的输入Tensor名称设置
    .fetch("output_tensor_name") // 根据模型的输出Tensor名称设置
    .run()
    .get(0);

float[] outputData = new float[outputTensor.shape()[1]];
outputTensor.copyTo(outputData);
// 对输出数据进行进一步处理

session.close();
graph.close();