torch.set_num_interop_threads

torch.set_num_interop_threads(int)

設定 CPU 上用於互操作並行（例如在 JIT 直譯器中）的執行緒數。

警告

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