파이썬 함수와 그 그라디언트를 텐서 흐름 연산으로 등록하려고합니다. 내가 많은 유용한 예 예 발견속성을 사용하여 파이썬에서 텐서 흐름 연산 정의하기
Write Custom Python-Based Gradient Function for an Operation? (without C++ Implementation) 그럼에도 불구하고
https://programtalk.com/python-examples/tensorflow.python.framework.function.Defun/
내가 작업의 속성을 등록하고 op.get_attr 호출하여 그라데이션 정의에 이러한 특성을 사용하고 싶은 ('ATTR_NAME') .
C 구현으로 이동하지 않고도이 작업을 수행 할 수 있습니까? 예를 들어 주시겠습니까?
안녕 피트와 감사합니다! 예를 들어 주시겠습니까? –