示例示例Java调用的方式有两种:使用 Java API,它是的一部分,可以在Java中直接调用的函数。
Java调用TensorFlow的方式有两种:
1. 使用TensorFlow Java API,它是TensorFlow的一部分,可以在Java中直接调用TensorFlow的函数。
代码示例:
// 创建TensorFlow图
Graph g = new Graph();
// 在图中定义常量
final Output a = g.constant("a", 3);
final Output b = g.constant("b", 2);
// 在图中定义计算
final Output c = g.add(a, b);
// 创建会话
Session sess = new Session(g);
// 运行计算
int result = sess.run(c);
System.out.println(result);
2. 使用TensorFlow Serving,它是一个独立的服务器,可以通过REST API或gRPC协议调用TensorFlow模型。
代码示例:
// 创建REST客户端
RestTemplate restTemplate = new RestTemplate();
// 调用TensorFlow Serving服务
String url = "http://localhost:8501/v1/models/my_model:predict";
Map
requestBody.put("instances", Arrays.asList(1.0, 2.0));
ResponseEntity
// 获取结果
Map
System.out.println(result);
本站系公益性非盈利分享网址,本文来自用户投稿,不代表边看边学立场,如若转载,请注明出处
评论列表(69条)