pyxu.info.deps

CUPY_ENABLED: bool

Show if CuPy-based backends are available.

class NDArrayInfo(value, names=<not given>, *values, module=None, qualname=None, type=None, start=1, boundary=None)

Bases: Enum

Supported dense array backends.

NUMPY = 1

DASK = 2

CUPY = 3

classmethod default()

Default array backend to use.

Return type:

NDArrayInfo

type()

Array type associated to a backend.

Return type:

type

classmethod from_obj(obj)

Find array backend associated to obj.

Return type:

NDArrayInfo

classmethod from_flag(gpu)

Find array backend suitable for in-memory CPU/GPU computing.

Parameters:

gpu (bool)

Return type:

NDArrayInfo

module(linalg=False)

Python module associated to an array backend.

Parameters:

linalg (bool) – Return the linear-algebra submodule with identical API to numpy.linalg.

Return type:

ModuleType

class SparseArrayInfo(value, names=<not given>, *values, module=None, qualname=None, type=None, start=1, boundary=None)

Bases: Enum

Supported sparse array backends.

SCIPY_SPARSE = 1

CUPY_SPARSE = 2

classmethod default()

Default array backend to use.

Return type:

SparseArrayInfo

type()

Array type associated to a backend.

Return type:

type

classmethod from_obj(obj)

Find array backend associated to obj.

Return type:

SparseArrayInfo

module(linalg=False)

Python module associated to an array backend.

Parameters:

linalg (bool) – Return the linear-algebra submodule with identical API to scipy.sparse.linalg.

Return type:

ModuleType

supported_array_types()

List of all supported dense array types in current Pyxu install.

Return type:

Collection[type]

supported_array_modules()

List of all supported dense array modules in current Pyxu install.

Return type:

Collection[ModuleType]

supported_sparse_types()

List of all supported sparse array types in current Pyxu install.

Return type:

Collection[type]

supported_sparse_modules()

List of all supported sparse array modules in current Pyxu install.

Return type:

Collection[ModuleType]