He

Ele inicializador.

Se a distribuição é TRUNCATED_NORMAL, extrai amostras a partir de uma distribuição normal truncada centrado em 0 com stddev = sqrt(2 / fanIn) onde fanIn é o número de unidades de entrada em peso do tensor.

Se a distribuição é uniforme, extrai amostras a partir de uma distribuição uniforme dentro [-limit, limit] , onde limit = sqrt(6 / fanIn) (FANIN é o número de unidades de entrada em peso do tensor).

Exemplos:

Ele 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);
 

Ele 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 um inicializador equivalente HeNormal, uso TRUNCATED_NORMAL para o parâmetro de distribuição.

Para um inicializador equivalente HeUniform, use UNIFORM para o parâmetro de distribuição.