Package ai.djl.nn

Class IdentityBlockFactory

java.lang.Object

ai.djl.nn.IdentityBlockFactory

All Implemented Interfaces:

BlockFactory, Serializable

public class IdentityBlockFactory
extends Object
implements BlockFactory

A BlockFactory class that creates IdentityBlock.

See Also:

• Serialized Form

Constructor Summary

Constructors

Constructor	Description
IdentityBlockFactory()

Method Summary

Modifier and Type	Method	Description
Block	newBlock(Model model, Path modelPath, Map<String,?> arguments)	Constructs the uninitialized block.

Methods inherited from class java.lang.Object

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

Constructor Details

IdentityBlockFactory

public IdentityBlockFactory()

Method Details

newBlock

public Block newBlock(Model model,
 Path modelPath,
 Map<String,?> arguments)

Constructs the uninitialized block.

Specified by:

newBlock in interface BlockFactory

Parameters:

model - the model of the block

modelPath - the directory of the model location

arguments - the block creation arguments

Returns:

the uninitialized block