torch.set_num_threads

torch.set_num_threads(int)

設定 CPU 上進行 intraop 平行計算所使用的執行緒數。

警告

為確保使用正確的執行緒數，必須在執行 eager、JIT 或 autograd 程式碼之前呼叫 set_num_threads。