tensor_wrapper module

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

Tools for converting between torch and numba-compatible arrays

class NumbaCompatibleTensorFunction

Bases: object

cpu_kernel()

launch_bounds(*shapes)

make_kernel()

out_shape(*shapes)

via_numpy(func): Decorator for piping a function through numpy arrays, and then giving the result back to torch.