PyTorch torch.get_float32_matmul_precision 函数
Pytorch torch 参考手册torch.get_float32_matmul_precision 是 PyTorch 中用于获取 FP32 矩阵乘法精度的函数。
函数定义
torch.get_float32_matmul_precision()
使用示例
实例
import torch
# 获取当前的 FP32 矩阵乘法精度设置
precision = torch.get_float32_matmul_precision()
print("FP32矩阵乘法精度:", precision)
# 获取当前的 FP32 矩阵乘法精度设置
precision = torch.get_float32_matmul_precision()
print("FP32矩阵乘法精度:", precision)
返回值说明：
- "highest" - 最高精度
- "high" - 高精度
- "medium" - 中等精度