Функция torch.set_default_dtype в PyTorch

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

torch.set_default_dtypeЭто функция в PyTorch, используемая для установки типа данных с плавающей запятой по умолчанию.

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

torch.set_default_dtype(d)

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

Пример

import torch

# Установить dtype по умолчанию в float64
torch.set_default_dtype(torch.float64)

# Создать тензор, используя float64 по умолчанию
x = torch.tensor([1.0, 2.0, 3.0])
print("Тип данных по умолчанию:", x.dtype)

# Восстановить настройки по умолчанию
torch.set_default_dtype(torch.float32)

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

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