Package ai.djl.nn

Class OnesBlockFactory

java.lang.Object

ai.djl.nn.OnesBlockFactory

All Implemented Interfaces:

BlockFactory, Serializable

public class OnesBlockFactory
extends Object
implements BlockFactory

A BlockFactory class that creates LambdaBlock.

See Also:

• Serialized Form

Constructor Summary

Constructors

Constructor	Description
OnesBlockFactory()

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

OnesBlockFactory

public OnesBlockFactory()

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