torch.set_num_interop_threads

torch.set_num_interop_threads(int)

在 CPU 上设置用于互操作并行性（例如，在 JIT 解释器中）的线程数。

警告

只能调用一次，并且必须在启动任何互操作并行工作（例如，JIT 执行）之前调用。