mlx.core.device_info

目錄

mlx.core.device_info#

device_info(d: Device = Device(cpu, 0)) dict[str, str | int]#

Get information about a device.

Returns a dictionary with device properties. Available keys depend on the backend and device type. Common keys include device_name, architecture, and total_memory (or memory_size).

參數:

d (Device) -- The device to query (defaults to the default device).

回傳:

Device information.

回傳型別:

dict