public class OrtSymbolBlock
extends ai.djl.nn.AbstractSymbolBlock
implements java.lang.AutoCloseable
OrtSymbolBlock is the ONNX Runtime implementation of SymbolBlock.
You can create a OrtSymbolBlock using Model.load(java.nio.file.Path,
String).
| Constructor and Description |
|---|
OrtSymbolBlock(ai.onnxruntime.OrtSession session,
OrtNDManager manager)
Constructs a
OrtSymbolBlock. |
| Modifier and Type | Method and Description |
|---|---|
void |
close() |
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) |
void |
removeLastBlock() |
addChildBlock, addParameter, beforeInitialize, cast, clear, describeInput, forward, forward, forwardInternal, getChildren, getDirectParameters, getParameters, initialize, initializeChildBlocks, isInitialized, loadMetadata, loadParameters, prepare, readInputShapes, saveInputShapes, saveMetadata, saveParameters, setInitializer, setInitializer, setInitializer, toStringpublic OrtSymbolBlock(ai.onnxruntime.OrtSession session,
OrtNDManager manager)
OrtSymbolBlock.
You can create a PtSymbolBlock using Model.load(java.nio.file.Path,
String).
session - the OrtSession contains the model informationmanager - the NDManager to holds the NDArraypublic void removeLastBlock()
removeLastBlock in interface ai.djl.nn.SymbolBlockprotected 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)
forwardInternal in class ai.djl.nn.AbstractBlockpublic void close()
close in interface java.lang.AutoCloseable