ai.djl.basicmodelzoo.basic

Class MlpBlockFactory

java.lang.Object

ai.djl.basicmodelzoo.basic.MlpBlockFactory

All Implemented Interfaces:

ai.djl.nn.BlockFactory, java.io.Serializable

public class MlpBlockFactory
extends java.lang.Object
implements ai.djl.nn.BlockFactory

A BlockFactory class that creates MLP block.

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
MlpBlockFactory()

Method Summary

Modifier and Type	Method and Description
ai.djl.nn.Block	newBlock(ai.djl.Model model, java.nio.file.Path modelPath, java.util.Map<java.lang.String,?> arguments)

Methods inherited from class java.lang.Object

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

Constructor Detail

MlpBlockFactory

public MlpBlockFactory()

Method Detail

newBlock

public ai.djl.nn.Block newBlock(ai.djl.Model model,
                                java.nio.file.Path modelPath,
                                java.util.Map<java.lang.String,?> arguments)

Specified by:

newBlock in interface ai.djl.nn.BlockFactory