Función PyTorch torch.get_float32_matmul_precision

Pytorch torch 参考手册Manual de referencia de Pytorch torch

torch.get_float32_matmul_precisionEs una función en PyTorch utilizada para obtener la precisión de la multiplicación de matrices FP32.

Definición de la función

torch.get_float32_matmul_precision()

Ejemplo de uso

Ejemplo

import torch

# Obtener la configuración actual de precisión de multiplicación de matrices FP32
precision = torch.get_float32_matmul_precision()
print("Precisión de multiplicación de matrices FP32:", precision)

Descripción del valor de retorno:

  • "highest" - precisión máxima
  • "high" - alta precisión
  • "medium" - precisión media

Pytorch torch 参考手册Manual de referencia de Pytorch torch

Otras extensiones