Package ai.djl.nn

Class IdentityBlockFactory

    • Method Summary

      All Methods Instance Methods Concrete Methods 
      Modifier and Type Method Description
      Block newBlock​(Model model, java.nio.file.Path modelPath, java.util.Map<java.lang.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 Detail

      • IdentityBlockFactory

        public IdentityBlockFactory()
    • 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:
        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