Missed TensorFlow Dev Summit? Check out the video playlist. Watch recordings

tf.keras.initializers.he_normal

View source on GitHub

He normal initializer.

tf.keras.initializers.he_normal(
    seed=None
)

It draws samples from a truncated normal distribution centered on 0 with stddev = sqrt(2 / fan_in) where fan_in is the number of input units in the weight tensor.

Arguments:

  • seed: A Python integer. Used to seed the random generator.

Returns:

An initializer.

References:

He et al., 2015 # pylint: disable=line-too-long (pdf)