PyTorch torch.set_num_threads Funktion

Pytorch torch 参考手册PyTorch torch-Referenzhandbuch

torch.set_num_threadsist eine Funktion in PyTorch, mit der die Anzahl der internen Threads festgelegt wird. Diese Threads werden verwendet, um Tensoroperationen parallel auszuführen (wie Matrixmultiplikation, Faltung usw.)

Funktionsdefinition

torch.set_num_threads(num_threads)

Parameterbeschreibung

  • num_threads: Die Anzahl der festzulegenden Threads

Anwendungsbeispiel

Beispiel

import torch

# Anzahl der Threads festlegen
torch.set_num_threads(4)

print("Anzahl der Threads eingestellt auf:", torch.get_num_threads())

# Große Matrizen zum Testen erstellen
a = torch.randn(1000, 1000)
b = torch.randn(1000, 1000)

# Matrixmultiplikation (verwendet die festgelegte Anzahl von Threads)
c = torch.matmul(a, b)

print("Matrixmultiplikation abgeschlossen, Form des Ergebnisses:", c.shape)

# Standardwerte wiederherstellen
torch.set_num_threads(1)

Das Ausgabeergebnis ist:

线程数已设置为: 4
矩阵乘法完成，结果形状: torch.Size([1000, 1000])

Pytorch torch 参考手册PyTorch torch-Referenzhandbuch

Andere Erweiterungen