ai.djl.nn

Interface SymbolBlock

All Superinterfaces:

Block

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 and Description
ai.djl.util.PairList<java.lang.String,Shape>	describeOutput() Returns a PairList of output names and shapes stored in model file.
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, getParameterShape, initialize, isInitialized, loadParameters, saveParameters, setInitializer, setInitializer, validateLayout

Method Detail

removeLastBlock

void removeLastBlock()

Removes the last block in the symbolic graph.

describeOutput

ai.djl.util.PairList<java.lang.String,Shape> describeOutput()

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

Returns:

the PairList of output names, and shapes