Функция PyTorch torch.get_float32_matmul_precision

Pytorch torch 参考手册Справочное руководство 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)

Описание возвращаемого значения:

  • "highest" - Наивысшая точность
  • "high" - Высокая точность
  • "medium" - Средняя точность

Pytorch torch 参考手册Справочное руководство Pytorch torch

Другие расширения