jax.Device

class jax.Device

可用裝置的描述符。

子類別用於表示特定類型的裝置，例如 CPU、GPU。子類別可能具有特定於該裝置類型的其他屬性。

__init__(*args, **kwargs)

方法

__init__(*args, **kwargs)

屬性

addressable_memories	傳回裝置可以定址的所有記憶體。
client	(self) -> object
default_memory	傳回裝置的預設記憶體。
device_kind	(self) -> str
get_stream_for_external_ready_events
host_id	已棄用；請使用 process_index
id	此裝置的整數 ID。
live_buffers
local_hardware_id	不透明硬體 ID，例如 CUDA 裝置編號。
memory
memory_stats	傳回此裝置的記憶體統計資訊，依名稱鍵控。
platform	(self) -> str
process_index	此裝置進程的整數索引。
task_id	已棄用；請使用 process_index
transfer_from_outfeed
transfer_to_infeed