Package ai.djl.nn.core

Class Linear.Builder

  • Enclosing class:
    Linear
    public static class Linear.Builder
extends java.lang.Object
    The Builder to construct a Linear type of Block.

    • Field Summary

      Fields 
      Modifier and Type Field Description
      protected long units  

    • Constructor Summary

      Constructors 
      Modifier Constructor Description
      protected Builder()  

    • Method Summary

      All Methods Instance Methods Concrete Methods 
      Modifier and Type Method Description
      Linear build()
      Returns the constructed Linear.
      Linear.Builder optBias​(boolean bias)
      Sets the optional parameter that indicates whether to include a bias vector with default value of true.
      Linear.Builder setUnits​(long units)
      Sets the number of output channels.

      • Methods inherited from class java.lang.Object

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

    • Field Detail

      • units

        protected long units

    • Constructor Detail

      • Builder

        protected Builder()

    • Method Detail

      • setUnits

        public Linear.Builder setUnits​(long units)
        Sets the number of output channels.
        Parameters:
        units - the number of desired output channels
        Returns:
        this Builder

      • optBias

        public Linear.Builder optBias​(boolean bias)
        Sets the optional parameter that indicates whether to include a bias vector with default value of true.
        Parameters:
        bias - whether to use a bias vector parameter
        Returns:
        this Builder

      • build

        public Linear build()
        Returns the constructed Linear.
        Returns:
        the constructed Linear
        Throws:
        java.lang.IllegalArgumentException - if all required parameters (outChannels) have not been set