ai.djl.nn

Class ParameterBlock

java.lang.Object

ai.djl.nn.AbstractBlock

ai.djl.nn.ParameterBlock

All Implemented Interfaces:

Block

Direct Known Subclasses:

BatchNorm, Convolution, Dropout, Embedding, LambdaBlock, Linear, Prelu, RecurrentCell

public abstract class ParameterBlock
extends AbstractBlock

ParameterBlock is an abstract implementation of Block. It is recommended that all Block classes that have no children extend the ParameterBlock.

Field Summary

Fields inherited from class ai.djl.nn.AbstractBlock

inputNames, inputShapes

Constructor Summary

Constructors

Constructor and Description
ParameterBlock()

Method Summary

Modifier and Type	Method and Description
BlockList	getChildren() Returns a list of all the children of the block.
Shape[]	initialize(NDManager manager, DataType dataType, Shape... inputShapes) Initializes the parameters of the block.
java.lang.String	toString()

Methods inherited from class ai.djl.nn.AbstractBlock

beforeInitialize, cast, clear, describeInput, getParameters, isInitialized, setInitializer, setInitializer

Methods inherited from class java.lang.Object

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

Methods inherited from interface ai.djl.nn.Block

forward, forward, getDirectParameters, getOutputShapes, getParameterShape, loadParameters, saveParameters, validateLayout

Constructor Detail

ParameterBlock

public ParameterBlock()

Method Detail

initialize

public Shape[] initialize(NDManager manager,
                          DataType dataType,
                          Shape... inputShapes)

Initializes the parameters of the block. This method must be called before calling `forward`.

Parameters:

manager - the NDManager to initialize the parameters

dataType - the datatype of the parameters

inputShapes - the shapes of the inputs to the block

Returns:

the shapes of the outputs of the block

getChildren

public final BlockList getChildren()

Returns a list of all the children of the block.

Returns:

the list of child blocks

toString

public java.lang.String toString()

Overrides:

toString in class java.lang.Object