torch.set_num_interop_threads

torch.set_num_interop_threads(int)

设置 CPU 上用于交互式并行（例如 JIT 解释器）的线程数。

警告

只能调用一次，并且必须在开始任何交互式并行工作（例如 JIT 执行）之前调用。