set_multithreading_enabled¶
- class torch.autograd.grad_mode.set_multithreading_enabled(mode)[source][source]¶
設定多執行緒反向傳播開啟或關閉的上下文管理器。
set_multithreading_enabled會根據其引數mode來啟用或停用多執行緒反向傳播。它可以作為一個上下文管理器或一個函式使用。這個上下文管理器是執行緒本地的；它不會影響其他執行緒中的計算。
- 引數
mode (bool) – 標誌，指示是否啟用多執行緒反向傳播 (
True) 或停用 (False)。
注意
這個 API 不適用於前向模式 AD。
