공개 최종 클래스 ApplyMomentum
모멘텀 체계에 따라 '*var'를 업데이트합니다.
Nesterov 추진력을 사용하려면 use_nesterov = True로 설정하십시오.
누적 = 누적 * 모멘텀 + 그라데이션 var -= lr * 누적
중첩 클래스
| 수업 | ApplyMomentum.옵션 | ApplyMomentum 의 선택적 속성 | |
상수
| 끈 | OP_NAME | TensorFlow 코어 엔진에서 알려진 이 작업의 이름 |
공개 방법
| 출력 <T> | 출력 () 텐서의 기호 핸들을 반환합니다. |
| static <T는 TType을 확장합니다. > ApplyMomentum <T> | |
| 출력 <T> | 밖으로 () "var"와 동일합니다. |
| 정적 ApplyMomentum.Options | useLocking (부울 useLocking) |
| 정적 ApplyMomentum.Options | useNesterov (부울 useNesterov) |
상속된 메서드
상수
공개 정적 최종 문자열 OP_NAME
TensorFlow 코어 엔진에서 알려진 이 작업의 이름
상수 값: "ApplyMomentum"
공개 방법
공개 출력 <T> asOutput ()
텐서의 기호 핸들을 반환합니다.
TensorFlow 작업에 대한 입력은 다른 TensorFlow 작업의 출력입니다. 이 메서드는 입력 계산을 나타내는 기호 핸들을 얻는 데 사용됩니다.
공개 정적 ApplyMomentum <T> 생성 ( 범위 범위, 피연산자 <T> var, 피연산자 <T> accum, 피연산자 <T> lr, 피연산자 <T> grad, 피연산자 <T> 추진력, 옵션... 옵션)
새로운 ApplyMomentum 작업을 래핑하는 클래스를 생성하는 팩토리 메서드입니다.
매개변수
| 범위 | 현재 범위 |
|---|---|
| var | Variable()에서 가져와야 합니다. |
| 축적 | Variable()에서 가져와야 합니다. |
| 난 | 배율 인수. 스칼라여야 합니다. |
| 졸업생 | 그라데이션입니다. |
| 기세 | 기세. 스칼라여야 합니다. |
| 옵션 | 선택적 속성 값을 전달합니다. |
보고
- ApplyMomentum의 새로운 인스턴스
공개 정적 ApplyMomentum.Options useLocking (부울 useLocking)
매개변수
| 사용잠금 | 'True'인 경우 var 및 accum 텐서 업데이트는 잠금으로 보호됩니다. 그렇지 않으면 동작이 정의되지 않지만 경합이 덜 나타날 수 있습니다. |
|---|
공개 정적 ApplyMomentum.Options useNesterov (부울 useNesterov)
매개변수
| 사용Nesterov | `True`인 경우, Compute grad에 전달된 텐서는 var - lr * Momentum * Accum이 됩니다. 따라서 결국 얻는 var는 실제로 var - lr * Momentum * Accum입니다. |
|---|