2017-11-09 1 views
1

1. C++로 MXNet에서 커스텀 연산자를 정의 할 때, C++에 정의 된 커스텀 연산자는 자동으로 파이썬 API를 어떻게 생성합니까? 2. 해당 코드는 어떻게 찾을 수 있습니까? 3. C++과 파이썬을 사용한 사용자 정의 연산자 정의의 차이점은 무엇입니까?MXNET C++ 코드는 python API와 어떻게 바인딩됩니까?

답변

1

매크로 NNVM_REGISTER_OP는 운영자를 이름으로 등록합니다.이 연산자는 파이썬에서 볼 수 있습니다. 거기에서 연산자의 속성 (즉, 호출 될 때 호출 할 함수 인 FCompute)을 정의합니다.

여기 C++ 섹션에는 가이드가 있습니다. https://mxnet.incubator.apache.org/how_to/new_op.html

관련 문제