Функция PyTorch torch.set_printoptions

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

torch.set_printoptionsЭто функция в PyTorch, используемая для настройки параметров печати тензоров.

Определение функции

torch.set_printoptions(precision, threshold, edgeitems, linewidth, profile, sci_mode)

Пример использования

Пример

import torch

# Установить точность печати до 2 знаков после запятой
torch.set_printoptions(precision=2)

x = torch.randn(3, 3)
print(x)

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

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