이 페이지는 Cloud Translation API를 통해 번역되었습니다.
Switch to English

tf.RegisterGradient

TensorFlow 1 개 버전 GitHub의에서 소스보기

연산 유형의 구배 함수를 등록하는 장식.

새로운 연산 유형을 정의 할 때이 장식에만 사용됩니다. 함께 OP 들어 m 입력과 n 출력 그라데이션 함수는 원래 얻어 함수 Operation 하고 n Tensor (연산의 각 출력에 대하여 기울기를 나타내는) 목적 및 복귀 m Tensor 와 부분 기울기를 나타내는 (개체 연산의 각 입력)에 대하여.

예를 들어, 가정의 동작 유형 것을 "Sub" 두 개의 입력 받아 xy , 단일 출력 리턴 x - y 다음 경사 함수를 등록 할 것이다 :

 @tf.RegisterGradient("Sub")
def _sub_grad(unused_op, grad):
  return grad, tf.negative(grad)
 

데코레이터 인수 op_type 작업의 문자열 유형입니다. 받는 사람이 해당한다 OpDef.name 동작을 정의하는 프로토에 대한 필드.

op_type 작업의 문자열 유형. 받는 사람이 해당한다 OpDef.name 동작을 정의하는 프로토에 대한 필드.

TypeError 경우 op_type 문자열이 아닙니다.

행동 양식

__call__

소스보기

레지스터 함수 f 그라데이션 함수 op_type .