공개 클래스 VarianceScaling
알려진 직접 서브클래스 |
가중치 텐서의 모양에 맞게 스케일을 조정할 수 있는 초기화 프로그램입니다.
distribution=TRUNCATED_NORMAL or NORMAL
사용하면 평균이 0이고 표준 편차가 있는 잘린/잘리지 않은 정규 분포(사용된 경우 잘린 후) stddev = Math.sqrt(scale / n)
에서 표본을 추출합니다. 여기서 n
은 다음과 같습니다.
-
mode=FAN_IN
인 경우 가중치 텐서의 입력 단위 수 -
mode=FAN_OUT
인 경우 출력 장치 수 -
mode=FAN_AVG
인 경우 입력 및 출력 장치 수의 평균
distribution=UNIFORM
사용하면 [-limit, limit]
내의 균일 분포에서 샘플이 추출됩니다. 여기서 limit = Math.sqrt(3 * scale / n);
.
예:
long seed = 1234l; float scale = 0.1f; VarianceScaling<TFloat32, TFloat32> initializer = new org.tensorflow.framework.initializers.VarianceScaling<>( tf, scale, Mode.FAN_IN, Distribution.UNIFORM, seed); Operand<TFloat32> values = initializer.call(tf.constant(Shape.of(2,2)), TFloat32.class);
중첩 클래스
열거형 | VarianceScaling.Distribution | 값을 초기화할 때 사용할 무작위 분포입니다. | |
열거형 | VarianceScaling.Mode | 팬 값을 계산하는 데 사용할 모드입니다. |
상수
더블 | SCALE_DEFAULT |
전지
공개 정적 최종 VarianceScaling.Distribution | DISTRIBUTION_DEFAULT | |
공개 정적 최종 VarianceScaling.Mode | MODE_DEFAULT |
공공 생성자
VarianceScaling (Ops tf, 긴 시드) VarianceScaling 초기화 프로그램을 생성합니다. | |
VarianceScaling (Ops tf, 이중 스케일, VarianceScaling.Mode 모드, VarianceScaling.Distribution 분포, 긴 시드) VarianceScaling 초기화 프로그램을 생성합니다. |
공개 방법
피연산자 <T> |
상속된 메서드
상수
공개 정적 최종 이중 SCALE_DEFAULT
상수값: 1.0
전지
공개 정적 최종 VarianceScaling.Distribution DISTRIBUTION_DEFAULT
공개 정적 최종 VarianceScaling.Mode MODE_DEFAULT
공공 생성자
공개 VarianceScaling (Ops tf, 긴 시드)
VarianceScaling 초기화 프로그램을 생성합니다.
매개변수
tf | TensorFlow 작업 |
---|---|
씨앗 | sed를 사용하여 임의의 시드를 생성합니다. |
공개 VarianceScaling (Ops tf, 이중 스케일, VarianceScaling.Mode 모드, VarianceScaling.Distribution 분포, 긴 시드)
VarianceScaling 초기화 프로그램을 생성합니다.
매개변수
tf | TensorFlow 작업 |
---|---|
규모 | 배율 인수(양수 부동 소수점). |
방법 | 분산 모드 |
분포 | 사용할 무작위 배포. |
씨앗 | 무작위 시드를 생성하는 데 사용됩니다. |