PyTorch torch.get_float32_matmul_precision Funktion

Pytorch torch 参考手册PyTorch torch Referenzhandbuch

torch.get_float32_matmul_precisionDies ist eine Funktion in PyTorch, mit der die Genauigkeit der FP32-Matrixmultiplikation abgerufen wird.

Funktionsdefinition

torch.get_float32_matmul_precision()

Anwendungsbeispiel

Beispiel

import torch

# Aktuelle Einstellung der FP32-Matrixmultiplikationsgenauigkeit abrufen
precision = torch.get_float32_matmul_precision()
print("FP32-Matrixmultiplikationsgenauigkeit:", precision)

Beschreibung des Rückgabewerts:

  • "highest" - Höchste Genauigkeit
  • "high" - Hohe Genauigkeit
  • "medium" - Mittlere Genauigkeit

Pytorch torch 参考手册PyTorch torch Referenzhandbuch

Andere Erweiterungen