Él inicializador.
Si la distribución es TRUNCATED_NORMAL, extrae 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, extrae 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:
El 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);
El 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 a HeNormal, utilice TRUNCATED_NORMAL
para el parámetro de distribución.
Para un inicializador equivalente a HeUniform, utilice UNIFORM
para el parámetro de distribución.
Constantes
doble | ESCALA |
Constantes heredadas
Campos heredados
Constructores Públicos
Métodos heredados
Constantes
ESCALA doble final estática pública
Constructores Públicos
público He (Ops tf, VarianceScaling.Distribución de distribución, semilla larga)
Crea un inicializador He
Parámetros
tf | las operaciones de TensorFlow |
---|---|
distribución | El tipo de distribución para el inicializador He. |
semilla | la semilla para la generación de números aleatorios. Un inicializador creado con una semilla determinada siempre producirá el mismo tensor aleatorio para una forma y un tipo determinados. |