tensor_wrapper

This is summary documentation for hippynn.custom_kernels.tensor_wrapper module. Click here for full documentation.

Links to Full Documentation:

Tools for converting between torch and numba-compatible arrays

Functions

via_numpy(func)

Decorator for piping a function through numpy arrays, and then giving the result back to torch.

Classes

NumbaCompatibleTensorFunction()