NumPy-CuPy Generic Code SupportΒΆ

cupy.get_array_module

Returns the array module for arguments.

cupyx.scipy.get_array_module

Returns the array module for arguments.