autograd_wrapper module

Full Documentation for hippynn.custom_kernels.autograd_wrapper module. Click here for a summary page.

Wraps non-pytorch implementations for use with pytorch autograd.

wrap_envops(envsum_impl, sensesum_impl, featsum_impl)

Parameters:

• envsum_impl – non-autograd implementation of envsum

• sensesum_impl – non-autograd implementation of sensesum

• featsum_impl – non-autograd implementtation of featsum

Returns: