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)

戻り値の説明：

  • 最高精度
  • 高精度
  • "medium" - 中程度の精度

Pytorch torch リファレンスマニュアル

その他の拡張