数値微分で実装するような書き方になってたな。
自動微分実装はさすがに一からはつらいわ。
tensorflowにしろpytorchにしろ結局一番めんどくさい実装はそこだしな。