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