ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

tf.compat.v1.keras.initializers.he_uniform

View source on GitHub

He uniform variance scaling initializer.

It draws samples from a uniform distribution within [-limit, limit] where limit is sqrt(6 / fan_in) where fan_in is the number of input units in the weight tensor.

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

An initializer.

References:

He et al., 2015

(pdf)