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 в качестве параметра распределения.

См. также

Константы

двойной ШКАЛА

Унаследованные константы

Унаследованные поля

Публичные конструкторы

Он (Ops tf, распределение VarianceScaling.Distribution , длинное начальное число)
Создает инициализатор He

Унаследованные методы

Константы

публичный статический финальный двойной SCALE

Постоянное значение: 2,0

Публичные конструкторы

public He (Ops tf, распределение VarianceScaling.Distribution , длинное начальное число)

Создает инициализатор He

Параметры
ТС Операции TensorFlow
распределение Тип распределения для инициализатора He.
семя начальное значение для генерации случайных чисел. Инициализатор, созданный с заданным начальным значением, всегда будет создавать один и тот же случайный тензор для заданной формы и типа d.
См. также