torch.set_num_interop_threads

torch.set_num_interop_threads(int)

設定 CPU 上用於互動式並行（例如 JIT 直譯器）的執行緒數。

警告

只能呼叫一次，並且必須在開始任何互動式並行工作（例如 JIT 執行）之前呼叫。