get_array_module

get_array_module#

mrinufft._array_compat.get_array_module(array)[source]#

Get the module of the array.

Parameters:

array (ndarray[tuple[int, ...], dtype[_ScalarType_co]])

Return type:

<module ‘numpy’ from ‘/volatile/github-ci-mind-inria/gpu_mind_runner/_work/mri-nufft/venv/lib/python3.10/site-packages/numpy/__init__.py’>