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。
