Package com.jml.neural_network.layers.initilizers

Class HeNormal

java.lang.Object
com.jml.neural_network.layers.initilizers.HeNormal
All Implemented Interfaces:
Initializer
public class HeNormal extends Object implements Initializer
layer parameter initializer to produce random values from a normal distribution with standard deviation std = sqrt(2 / fanIn) and mean of 0 where fanIn is the input dimension for the layer.

  • Constructor Summary

    Constructors
    Constructor
    Description
    HeNormal()
    Creates a HeNormal initializer.
    HeNormal​(long seed)
    Creates a HeNormal 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

    • HeNormal

      public HeNormal()
      Creates a HeNormal initializer.

    • HeNormal

      public HeNormal​(long seed)
      Creates a HeNormal 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.