public abstract class AbstractSymbolBlock extends AbstractBlock implements SymbolBlock
AbstractSymbolBlock
is an abstract implementation of SymbolBlock
.children, inputNames, inputShapes, parameters, version
Constructor and Description |
---|
AbstractSymbolBlock(byte version)
Builds an empty block with the given version for parameter serialization.
|
Modifier and Type | Method and Description |
---|---|
Shape[] |
getOutputShapes(Shape[] inputShapes)
Returns the expected output shapes of the block for the specified input shapes.
|
addChildBlock, addParameter, beforeInitialize, cast, clear, describeInput, forward, forward, forwardInternal, forwardInternal, getChildren, getDirectParameters, getParameters, initialize, initializeChildBlocks, isInitialized, loadMetadata, loadParameters, prepare, readInputShapes, saveInputShapes, saveMetadata, saveParameters, setInitializer, setInitializer, setInitializer, toString
clone, equals, finalize, getClass, hashCode, notify, notifyAll, wait, wait, wait
describeOutput, newInstance, removeLastBlock
cast, clear, describeInput, forward, forward, forward, getChildren, getDirectParameters, getParameters, initialize, isInitialized, loadParameters, saveParameters, setInitializer, setInitializer, setInitializer, validateLayout
public AbstractSymbolBlock(byte version)
version
- the version to use for parameter serialization.public Shape[] getOutputShapes(Shape[] inputShapes)
getOutputShapes
in interface Block
inputShapes
- the shapes of the inputs