torch.set_num_interop_threads

torch.set_num_interop_threads(int)

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

警告

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