Connecting PyMC3 to external code - help with understanding Theano custom Ops

Hi, you need 2 Ops, first for forward, 2d for backward.

Good example is
Eigh forward https://github.com/Theano/Theano/blob/master/theano/tensor/nlinalg.py#L293
Eigh backward https://github.com/Theano/Theano/blob/master/theano/tensor/nlinalg.py#L365

1 Like