Java 如何为单个浮点值创建ONNXTensor?
我有以下java代码:Java 如何为单个浮点值创建ONNXTensor?,java,onnx,onnxruntime,onnx-coreml,Java,Onnx,Onnxruntime,Onnx Coreml,我有以下java代码: try (OrtEnvironment env = OrtEnvironment.getEnvironment(); OrtSession.SessionOptions opts = new OrtSession.SessionOptions()) { opts.setOptimizationLevel(OrtSession.SessionOptions.OptLevel.BASIC_OPT);
try (OrtEnvironment env = OrtEnvironment.getEnvironment();
OrtSession.SessionOptions opts = new OrtSession.SessionOptions()) {
opts.setOptimizationLevel(OrtSession.SessionOptions.OptLevel.BASIC_OPT);
try (OrtSession session = env.createSession("model.onnx", opts)) {
OnnxTensor.createTensor(env, 10.0f);
}
}
我想创建一个浮点值的OnnxTensor。我得到以下例外情况:
Exception in thread "main" ai.onnxruntime.OrtException: Cannot convert class java.lang.Float to an OnnxTensor.
at ai.onnxruntime.TensorInfo.constructFromJavaArray(TensorInfo.java:208)
at ai.onnxruntime.OnnxTensor.createTensor(OnnxTensor.java:330)
at ai.onnxruntime.OnnxTensor.createTensor(OnnxTensor.java:314)