Package ai.djl.nn

Interface SymbolBlock

All Superinterfaces:

Block

All Known Implementing Classes:

AbstractSymbolBlock

public interface SymbolBlock
extends Block

SymbolBlock is a Block is used to load models that were exported directly from the engine in its native format.

Method Summary

Modifier and Type	Method	Description
default ai.djl.util.PairList<String,Shape>	describeOutput()	Returns a PairList 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, freezeParameters, freezeParameters, getChildren, getDirectParameters, getInputShapes, getOutputDataTypes, getOutputShapes, getOutputShapes, getParameters, initialize, isInitialized, loadParameters, saveParameters, setInitializer, setInitializer, setInitializer

Method Details

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<String,Shape> describeOutput()

Returns a PairList of output names and shapes stored in model file.

Returns:

the PairList of output names, and shapes