jax.transfer_guard#

jax.transfer_guard(new_val)[源代码]#

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

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

参数:

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

产生:

无。

返回类型:

Iterator[None]