He

공개 수업 그는

그는 초기화 프로그램입니다.

분포가 TRUNCATED_NORMAL인 경우 stddev = sqrt(2 / fanIn) 사용하여 0 중심으로 하는 잘린 정규 분포에서 샘플을 그립니다. 여기서 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 사용합니다.

참조

상수

더블 규모

상속된 상수

상속된 필드

공공 생성자

He (Ops tf, VarianceScaling.Distribution 분포, 긴 시드)
He 초기화 프로그램을 생성합니다.

상속된 메서드

상수

공개 정적 최종 이중 SCALE

상수값: 2.0

공공 생성자

공개 He (Ops tf, VarianceScaling.Distribution 분포, 긴 시드)

He 초기화 프로그램을 생성합니다.

매개변수
tf TensorFlow 작업
분포 He 이니셜라이저의 배포 유형입니다.
씨앗 난수 생성을 위한 시드입니다. 주어진 시드로 생성된 초기화는 항상 주어진 모양과 dtype에 대해 동일한 무작위 텐서를 생성합니다.
참조