Package ai.djl.nn
Interface SymbolBlock
-
- All Superinterfaces:
Block
- All Known Implementing Classes:
AbstractSymbolBlock
public interface SymbolBlock extends Block
SymbolBlock
is aBlock
is used to load models that were exported directly from the engine in its native format.
-
-
Method Summary
All Methods Static Methods Instance Methods Default Methods Modifier and Type Method Description default ai.djl.util.PairList<java.lang.String,Shape>
describeOutput()
Returns aPairList
of output names and shapes stored in model file.static SymbolBlock
newInstance(NDManager manager)
Creates an empty SymbolBlock instance.default void
removeLastBlock()
Removes the last block in the symbolic graph.-
Methods inherited from interface ai.djl.nn.Block
cast, clear, describeInput, forward, forward, forward, getChildren, getDirectParameters, getOutputShapes, getParameters, initialize, isInitialized, loadParameters, saveParameters, setInitializer, setInitializer, setInitializer
-
-
-
-
Method Detail
-
newInstance
static SymbolBlock newInstance(NDManager manager)
Creates an empty SymbolBlock instance.- Parameters:
manager
- the manager to be applied in the SymbolBlock- Returns:
- a new Model instance
-
removeLastBlock
default void removeLastBlock()
Removes the last block in the symbolic graph.
-
describeOutput
default ai.djl.util.PairList<java.lang.String,Shape> describeOutput()
Returns aPairList
of output names and shapes stored in model file.- Returns:
- the
PairList
of output names, and shapes
-
-