jax.process_index

jax.process_index(backend=None)

傳回此進程的整數進程索引。

在大多數平台上，這將始終為 0。但在多進程平台上，這會有所不同。

參數:

backend (str | xla_client.Client | None | None) – 這是一個實驗性功能，API 可能會變更。選填，一個表示 xla 後端的字串：'cpu'、'gpu' 或 'tpu'。

傳回值:

整數進程索引。

傳回類型:

int