He

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

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