He

Он инициализатор.

Если распределение TRUNCATED_NORMAL, оно извлекает выборки из усеченного нормального распределения с центром в 0 с помощью stddev = sqrt(2 / fanIn) где fanIn — количество входных единиц в тензоре веса.

Если распределение UNIFORM, оно извлекает образцы из равномерного распределения в пределах [-limit, limit] , где limit = sqrt(6 / fanIn) (fanIn — это количество входных единиц в тензоре веса).

Примеры:

Он нормальный:

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

Он униформа:

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

ПРИМЕЧАНИЕ:

Для эквивалентного инициализатора HeNormal используйте TRUNCATED_NORMAL в качестве параметра распределения.

Для эквивалентного инициализатора HeUniform используйте UNIFORM в качестве параметра распределения.