É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. |