ai.djl.nn

Class AbstractSymbolBlock

java.lang.Object

ai.djl.nn.AbstractBlock

ai.djl.nn.AbstractSymbolBlock

All Implemented Interfaces:

Block, SymbolBlock

public abstract class AbstractSymbolBlock
extends AbstractBlock
implements SymbolBlock

AbstractSymbolBlock is an abstract implementation of SymbolBlock.

Field Summary

Fields inherited from class ai.djl.nn.AbstractBlock

children, inputNames, inputShapes, parameters, parameterShapeCallbacks, version

Constructor Summary

Constructors

Constructor and Description
AbstractSymbolBlock(byte version) Builds an empty block with the given version for parameter serialization.

Method Summary

Modifier and Type	Method and Description
Shape[]	getOutputShapes(NDManager manager, Shape[] inputShapes) Returns the expected output shapes of the block for the specified input shapes.

Methods inherited from class ai.djl.nn.AbstractBlock

addChildBlock, addParameter, addParameter, addParameter, beforeInitialize, cast, clear, describeInput, forward, forwardInternal, getChildren, getDirectParameters, getParameters, getParameterShape, initialize, initializeChildBlocks, isInitialized, loadMetadata, loadParameters, readInputShapes, saveInputShapes, saveMetadata, saveParameters, setInitializer, setInitializer, toString

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, wait, wait, wait

Methods inherited from interface ai.djl.nn.SymbolBlock

describeOutput, removeLastBlock

Methods inherited from interface ai.djl.nn.Block

cast, clear, describeInput, forward, forward, forward, getChildren, getDirectParameters, getParameters, getParameterShape, initialize, isInitialized, loadParameters, saveParameters, setInitializer, setInitializer, validateLayout

Constructor Detail

AbstractSymbolBlock

public AbstractSymbolBlock(byte version)

Builds an empty block with the given version for parameter serialization.

Parameters:

version - the version to use for parameter serialization.

Method Detail

getOutputShapes

public Shape[] getOutputShapes(NDManager manager,
                               Shape[] inputShapes)

Returns the expected output shapes of the block for the specified input shapes.

Specified by:

getOutputShapes in interface Block

Parameters:

manager - an NDManager

inputShapes - the shapes of the inputs

Returns:

the expected output shapes of the block