jax.transfer_guard

jax.transfer_guard(new_val)

一个上下文管理器，用于控制所有传输的传输保护级别。

有关更多信息，请参阅 https://jax.ac.cn/en/latest/transfer_guard.html

参数:

new_val (str) – 所有传输的新的线程本地传输保护级别。

产量:

None.

返回类型:

Iterator[None]