PyTorch torch.get_num_interop_threads 関数

Pytorch torch 参考手册Pytorch torch リファレンスマニュアル

torch.get_num_interop_threadsPyTorch で inter-op（オペレーション間）スレッド数を取得するために使用される関数です。

関数定義

torch.get_num_interop_threads()

使用例

実例

import torch

# 現在の inter-op スレッド数を取得する
num_threads = torch.get_num_interop_threads()
print("inter-op スレッド数:", num_threads)

Pytorch torch 参考手册Pytorch torch リファレンスマニュアル

その他の拡張