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