Class FeedForwardLayer.Builder<T extends FeedForwardLayer.Builder<T>>

    • Field Detail

      • nIn

        protected long nIn
        Number of inputs for the layer (usually the size of the last layer).
        Note that for Convolutional layers, this is the input channels, otherwise is the previous layer size.
      • nOut

        protected long nOut
        Number of inputs for the layer (usually the size of the last layer).
        Note that for Convolutional layers, this is the input channels, otherwise is the previous layer size.
    • Constructor Detail

      • Builder

        public Builder()
    • Method Detail

      • nIn

        public T nIn​(int nIn)
        Number of inputs for the layer (usually the size of the last layer).
        Note that for Convolutional layers, this is the input channels, otherwise is the previous layer size.
        Parameters:
        nIn - Number of inputs for the layer
      • nIn

        public T nIn​(long nIn)
        Number of inputs for the layer (usually the size of the last layer).
        Note that for Convolutional layers, this is the input channels, otherwise is the previous layer size.
        Parameters:
        nIn - Number of inputs for the layer
      • nOut

        public T nOut​(int nOut)
        Number of outputs - used to set the layer size (number of units/nodes for the current layer). Note that this is equivalent to units(int)
        Parameters:
        nOut - Number of outputs / layer size
      • nOut

        public T nOut​(long nOut)
        Number of outputs - used to set the layer size (number of units/nodes for the current layer). Note that this is equivalent to units(int)
        Parameters:
        nOut - Number of outputs / layer size
      • units

        public T units​(int units)
        Set the number of units / layer size for this layer.
        This is equivalent to nOut(int)
        Parameters:
        units - Size of the layer (number of units) / nOut
        See Also:
        nOut(int)