cupyx.scipy.get_array_module#
- cupyx.scipy.get_array_module(*args)[源代码][源代码]#
返回参数的数组模块。
此函数用于实现 CPU/GPU 通用代码。如果至少有一个参数是
cupy.ndarray对象，则返回cupyx.scipy模块。- 参数:
args – 用于确定是使用 NumPy 还是 CuPy 的值。
- 返回:
cupyx.scipy或scipy根据参数的类型返回。- 返回类型:
module
返回参数的数组模块。
此函数用于实现 CPU/GPU 通用代码。如果至少有一个参数是 cupy.ndarray 对象，则返回 cupyx.scipy 模块。
args – 用于确定是使用 NumPy 还是 CuPy 的值。
cupyx.scipy 或 scipy 根据参数的类型返回。
module