Package ai.djl.nn
Class IdentityBlockFactory
- java.lang.Object
-
- ai.djl.nn.IdentityBlockFactory
-
- All Implemented Interfaces:
BlockFactory,java.io.Serializable
public class IdentityBlockFactory extends java.lang.Object implements BlockFactory
ABlockFactoryclass that creates IdentityBlock.- See Also:
- Serialized Form
-
-
Constructor Summary
Constructors Constructor Description IdentityBlockFactory()
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description BlocknewBlock(Model model, java.nio.file.Path modelPath, java.util.Map<java.lang.String,?> arguments)Constructs the uninitialized block.
-
-
-
Method Detail
-
newBlock
public Block newBlock(Model model, java.nio.file.Path modelPath, java.util.Map<java.lang.String,?> arguments)
Constructs the uninitialized block.- Specified by:
newBlockin interfaceBlockFactory- Parameters:
model- the model of the blockmodelPath- the directory of the model locationarguments- the block creation arguments- Returns:
- the uninitialized block
-
-