Class ONNXRuntimeBackend.ONNXSession

java.lang.Object

org.episteme.core.mathematics.ml.neural.backends.ONNXRuntimeBackend.ONNXSession

Enclosing class:

ONNXRuntimeBackend

public static class ONNXRuntimeBackend.ONNXSession
extends Object

Represents a loaded ONNX model session.

Since:

2.0

Author:

Silvere Martin-Michiellot, Gemini AI (Google DeepMind)

Constructor Summary

Constructors

Constructor	Description
ONNXSession(Path path, Object nativeSession)

Method Summary

Modifier and Type	Method	Description
Map<String, Tensor<?>>	run(Map<String, Tensor<?>> inputs)	Runs the model with input tensors.

Methods inherited from class Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Details

ONNXSession

public ONNXSession(Path path,
 Object nativeSession)

Method Details

run

public Map<String, Tensor<?>> run(Map<String, Tensor<?>> inputs)

Runs the model with input tensors.

Parameters:

inputs - map of input name to Tensor.

Returns:

map of output name to Tensor.