El inicializador.
Si la distribución es TRUNCATED_NORMAL, señala a muestras de una distribución normal truncada centrada en 0
con stddev = sqrt(2 / fanIn)
donde fanIn
es el número de unidades de entrada en el tensor de peso.
Si la distribución es uniforme, se basa muestras de una distribución uniforme dentro de [-limit, limit]
, donde limit = sqrt(6 / fanIn)
(Fanin es el número de unidades de entrada en el tensor de peso).
Ejemplos:
Él normal:
long seed = 1001l; He<TFloat32, TFloat32> initializer = new org.tensorflow.framework.initializers.He<>(tf, Distribution.TRUNCATED_NORMAL, seed);); Operand<TFloat32> values = initializer.call(tf.constant(Shape.of(2,2)), TFloat32.class);
Él Uniforme:
long seed = 1001l; He<TFloat32, TFloat32> initializer = new org.tensorflow.framework.initializers.He<>(tf, Distribution.UNIFORM, seed);); Operand<TFloat32> values = initializer.call(tf.constant(Shape.of(2,2)), TFloat32.class);
NOTA:
Para un inicializador equivalente HeNormal, el uso TRUNCATED_NORMAL
para el parámetro de distribución.
Para un inicializador equivalente HeUniform, utilice UNIFORM
para el parámetro de distribución.
Constantes
doble | ESCALA |
Constantes heredadas
Campos heredados
Constructores públicos
Métodos heredados
Constantes
Doble escala public static final
Constructores públicos
Él pública (OPS tf, VarianceScaling.Distribution distribución, larga semilla)
Crea un inicializador He
Parámetros
tf | las operaciones de TensorFlow |
---|---|
distribución | El tipo de distribución del inicializador He. |
semilla | la semilla para la generación de números aleatorios. Un inicializador creado con una semilla dada siempre producirá el mismo tensor aleatorio para una forma y tipo d dados. |