`heat._array_api_inspection`#

Module Contents#

class __array_namespace_info__#

Get the array API inspection namespace for Heat.

capabilities()[source]#: Returns a dictionary of array library capabilities.

default_device()[source]#: Returns the default device.

default_dtypes(*, device=None)[source]#: Returns a dictionary containing default data types.

devices()[source]#: Returns a list of supported devices which are available at runtime.

dtypes(*, device=None, kind=None)[source]#: Returns a dictionary of supported Array API data types