안녕하세요
지금 원하시는 함수가 외부에 있어서 numpy로 변경을 하시는 것 같습니다.
아시는 바와 같이 이 경우에 자동 미분이 동작하지 않기 때문에 가능하시면 텐서로 해당 기능을 구현하시는 것이 좋습니다.
부득이하게 연산을 파이토치 외에서 하게 될 경우에는 로스에서 해당 함수 이전까지 그레디언트를 직접 계산해서 전달해주어야 합니다.
제가 알고 있는 이런 예제들은 CTC, transducer Loss 같은 것들이 있는데요. C++로 함수를 구현하고 그레디언트를 직접계산해서 파이토치 텐서 형태로 포맷을 해서 전달합니다. 사실 cudnn 함수들도 이런 식으로 구현이 되어 있는 것이지요.