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
,java.lang.AutoCloseable
public class OrtSymbolBlock extends ai.djl.nn.AbstractSymbolBlock implements java.lang.AutoCloseable
OrtSymbolBlock
is the ONNX Runtime implementation ofSymbolBlock
.You can create a
OrtSymbolBlock
usingModel.load(java.nio.file.Path, String)
.
-
-
Constructor Summary
Constructors Constructor Description OrtSymbolBlock(ai.onnxruntime.OrtSession session, OrtNDManager manager)
Constructs aOrtSymbolBlock
.
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description void
close()
ai.djl.util.PairList<java.lang.String,ai.djl.ndarray.types.Shape>
describeInput()
protected ai.djl.ndarray.NDList
forwardInternal(ai.djl.training.ParameterStore parameterStore, ai.djl.ndarray.NDList inputs, boolean training, ai.djl.util.PairList<java.lang.String,java.lang.Object> params)
ai.djl.nn.ParameterList
getDirectParameters()
void
removeLastBlock()
-
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
-
-
-
-
Constructor Detail
-
OrtSymbolBlock
public OrtSymbolBlock(ai.onnxruntime.OrtSession session, OrtNDManager manager)
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 Detail
-
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<java.lang.String,java.lang.Object> params)
- Specified by:
forwardInternal
in classai.djl.nn.AbstractBaseBlock
-
describeInput
public ai.djl.util.PairList<java.lang.String,ai.djl.ndarray.types.Shape> describeInput()
- Specified by:
describeInput
in interfaceai.djl.nn.Block
- Overrides:
describeInput
in classai.djl.nn.AbstractBaseBlock
-
close
public void close()
- Specified by:
close
in interfacejava.lang.AutoCloseable
-
getDirectParameters
public ai.djl.nn.ParameterList getDirectParameters()
- Specified by:
getDirectParameters
in interfaceai.djl.nn.Block
-
-