He

Lui inizializzatore.

Se la distribuzione è TRUNCATED_NORMAL, trae campioni da una distribuzione normale troncata centrata su 0 con stddev = sqrt(2 / fanIn) dove fanIn è il numero di unità di input nel tensore del peso.

Se la distribuzione è UNIFORM, trae campioni da una distribuzione uniforme entro [-limit, limit] , dove limit = sqrt(6 / fanIn) (fanIn è il numero di unità di input nel tensore del peso).

Esempi:

Lui normale:

     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:

Per un inizializzatore equivalente HeNormal, utilizzare TRUNCATED_NORMAL per il parametro di distribuzione.

Per un inizializzatore equivalente HeUniform, utilizzare UNIFORM per il parametro di distribuzione.