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[source]

Bases: object

cpu_kernel()[source]

launch_bounds(*shapes)[source]

make_kernel()[source]

out_shape(*shapes)[source]

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