PyTorch torch.get_float32_matmul_precision Function

Pytorch torch 参考手册Pytorch torch Reference Manual

torch.get_float32_matmul_precisionThis is a function in PyTorch used to get the FP32 matrix multiplication precision.

Function Definition

torch.get_float32_matmul_precision()

Usage Example

Example

import torch

# Get the current FP32 matrix multiplication precision setting
precision = torch.get_float32_matmul_precision()
print("FP32 matrix multiplication precision:", precision)

Return value description:

  • "highest" - Highest precision
  • "high" - High precision
  • "medium" - Medium precision

Pytorch torch 参考手册Pytorch torch Reference Manual

Other Extensions