jax.transfer_guard

jax.transfer_guard(new_val)

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

更多信息请参阅 https://jax.net.cn/en/latest/transfer_guard.html

参数:

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

返回值:

无。

返回类型:

Iterator[None]