jax.transfer_guard#

jax.transfer_guard(new_val)[原始碼]#

用於控制所有傳輸的 transfer guard 等級的上下文管理器。

更多資訊，請參閱 https://jax.dev.org.tw/en/latest/transfer_guard.html

參數:

new_val (str) – 所有傳輸的新執行緒本地 transfer guard 等級。

Yields:

None.

返回類型:

Iterator[None]