Package ai.djl.onnxruntime.engine
Class OrtSymbolBlock
java.lang.Object
ai.djl.nn.AbstractBaseBlock
ai.djl.nn.AbstractSymbolBlock
ai.djl.onnxruntime.engine.OrtSymbolBlock
- All Implemented Interfaces:
ai.djl.nn.Block
,ai.djl.nn.SymbolBlock
,AutoCloseable
OrtSymbolBlock
is the ONNX Runtime implementation of SymbolBlock
.
You can create a OrtSymbolBlock
using Model.load(java.nio.file.Path, String)
.
-
Field Summary
Fields inherited from class ai.djl.nn.AbstractBaseBlock
inputNames, inputShapes, outputDataTypes, version
-
Constructor Summary
ConstructorDescriptionOrtSymbolBlock
(ai.onnxruntime.OrtSession session, OrtNDManager manager) Constructs aOrtSymbolBlock
. -
Method Summary
Modifier and TypeMethodDescriptionvoid
close()
ai.djl.util.PairList<String,
ai.djl.ndarray.types.Shape> protected ai.djl.ndarray.NDList
forwardInternal
(ai.djl.training.ParameterStore parameterStore, ai.djl.ndarray.NDList inputs, boolean training, ai.djl.util.PairList<String, Object> params) ai.djl.nn.ParameterList
void
Methods inherited from class ai.djl.nn.AbstractSymbolBlock
getChildren, getOutputShapes
Methods inherited from class ai.djl.nn.AbstractBaseBlock
beforeInitialize, cast, clear, forward, forward, forwardInternal, getInputShapes, getOutputDataTypes, getParameters, initialize, initializeChildBlocks, isInitialized, loadMetadata, loadParameters, prepare, readInputShapes, saveInputShapes, saveMetadata, saveParameters, setInitializer, setInitializer, setInitializer, toString
Methods inherited from class java.lang.Object
clone, equals, finalize, getClass, hashCode, notify, notifyAll, wait, wait, wait
Methods inherited from interface ai.djl.nn.Block
cast, clear, forward, forward, forward, freezeParameters, freezeParameters, getInputShapes, getOutputDataTypes, getOutputShapes, getParameters, initialize, isInitialized, loadParameters, saveParameters, setInitializer, setInitializer, setInitializer
Methods inherited from interface ai.djl.nn.SymbolBlock
describeOutput
-
Constructor Details
-
OrtSymbolBlock
Constructs aOrtSymbolBlock
.You can create a
PtSymbolBlock
usingModel.load(java.nio.file.Path, String)
.- Parameters:
session
- theOrtSession
contains the model informationmanager
- theNDManager
to holds the NDArray
-
-
Method Details
-
removeLastBlock
public void removeLastBlock()- Specified by:
removeLastBlock
in interfaceai.djl.nn.SymbolBlock
-
forwardInternal
protected ai.djl.ndarray.NDList forwardInternal(ai.djl.training.ParameterStore parameterStore, ai.djl.ndarray.NDList inputs, boolean training, ai.djl.util.PairList<String, Object> params) - Specified by:
forwardInternal
in classai.djl.nn.AbstractBaseBlock
-
describeInput
- Specified by:
describeInput
in interfaceai.djl.nn.Block
- Overrides:
describeInput
in classai.djl.nn.AbstractBaseBlock
-
close
public void close()- Specified by:
close
in interfaceAutoCloseable
-
getDirectParameters
public ai.djl.nn.ParameterList getDirectParameters()- Specified by:
getDirectParameters
in interfaceai.djl.nn.Block
-