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