Class MlpBlockFactory

java.lang.Object
ai.djl.basicmodelzoo.basic.MlpBlockFactory
All Implemented Interfaces:
ai.djl.nn.BlockFactory, Serializable

public class MlpBlockFactory extends Object implements ai.djl.nn.BlockFactory
A BlockFactory class that creates MLP block.
See Also:
  • Constructor Details

    • MlpBlockFactory

      public MlpBlockFactory()
  • Method Details

    • newBlock

      public ai.djl.nn.Block newBlock(ai.djl.Model model, Path modelPath, Map<String,?> arguments)
      Specified by:
      newBlock in interface ai.djl.nn.BlockFactory