set_multithreading_enabled¶
- class torch.autograd.grad_mode.set_multithreading_enabled(mode)[source][source]¶
设置多线程反向传播开启或关闭的上下文管理器。
set_multithreading_enabled会根据其参数mode来启用或禁用多线程反向传播。它可以作为一个上下文管理器或一个函数使用。这个上下文管理器是线程本地的；它不会影响其他线程中的计算。
- 参数
mode (bool) – 标志，指示是否启用多线程反向传播 (
True) 或禁用 (False)。
注意
这个 API 不适用于前向模式 AD。
