jax.Device

class jax.Device

可用设备的描述符。

子类用于表示特定类型的设备，例如 CPU、GPU。子类可能具有该设备类型特有的额外属性。

__init__(*args, **kwargs)

方法

__init__(*args, **kwargs)

属性

可寻址内存	返回设备可寻址的所有内存。
客户端	(self) -> object
默认内存	返回设备的默认内存。
设备类型	(self) -> str
获取外部就绪事件的流
主机ID	已弃用；请使用 process_index
ID	此设备的整数ID。
活跃缓冲区
本地硬件ID	不透明硬件ID，例如 CUDA 设备编号。
内存
内存统计	返回按名称键控的此设备的内存统计信息。
平台	(self) -> str
进程索引	此设备进程的整数索引。
任务ID	已弃用；请使用 process_index