Fonction PyTorch torch.get_float32_matmul_precision

Pytorch torch 参考手册Manuel de référence Pytorch torch

torch.get_float32_matmul_precisionCette fonction de PyTorch permet d'obtenir la précision de multiplication matricielle FP32.

Définition de fonction

torch.get_float32_matmul_precision()

Exemple d'utilisation

Exemple

import torch

# Obtenir le réglage actuel de la précision de multiplication matricielle FP32
precision = torch.get_float32_matmul_precision()
print("Précision de multiplication matricielle FP32 :", precision)

Description de la valeur de retour :

  • "highest" - Précision maximale
  • "high" - Précision élevée
  • "medium" - Précision moyenne

Pytorch torch 参考手册Manuel de référence Pytorch torch

Autres extensions