Package com.jml.neural_network.layers.initilizers

Class HeUniform

java.lang.Object
com.jml.neural_network.layers.initilizers.HeUniform
All Implemented Interfaces:
Initializer
public class HeUniform extends Object implements Initializer
layer parameter initializer to produce random values from a clipped uniform distribution in [-lim, lim] where lim = sqrt(6 / fanIn), fanIn is the input dimension for the layer.

  • Constructor Summary

    Constructors
    Constructor
    Description
    HeUniform()
    Creates a HeUniform initializer.
    HeUniform​(long seed)
    Creates a HeUniform initializer.

  • Method Summary

    Modifier and Type
    Method
    Description
    linalg.Matrix
    init​(int m, int n)
    Initializes values of a weight/bias matrix as specified by this initializer.

    Methods inherited from class java.lang.Object

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

  • Constructor Details

    • HeUniform

      public HeUniform()
      Creates a HeUniform initializer.

    • HeUniform

      public HeUniform​(long seed)
      Creates a HeUniform initializer.
      Parameters:
      seed - Seed for random number generator.

  • Method Details

    • init

      public linalg.Matrix init​(int m, int n)
      Initializes values of a weight/bias matrix as specified by this initializer.
      Specified by:
      init in interface Initializer
      Parameters:
      m - Number of rows in the matrix.
      n - Number of columns in the matrix.
      Returns:
      A Matrix initialized with values from the distribution specified by this initializer.