PyTorch torch.get_num_interop_threads 函数
torch.get_num_interop_threads 是 PyTorch 中用于获取跨域(inter-op)线程数的函数。
函数定义
torch.get_num_interop_threads()
使用示例
实例
import torch
# 获取当前的跨域线程数
num_threads = torch.get_num_interop_threads()
print("跨域线程数:", num_threads)
# 获取当前的跨域线程数
num_threads = torch.get_num_interop_threads()
print("跨域线程数:", num_threads)

Pytorch torch 参考手册