torch.set_default_dtype

torch.set_default_dtype(d) [source]

Устанавливает значение по умолчанию для плавающей точки типа данных на d. Поддерживает torch.float32 и torch.float64 в качестве входных данных. Другие типы данных могут приниматься без замечаний, но не поддерживаются и вряд ли будут работать так, как ожидается.

При инициализации PyTorch значение по умолчанию для плавающего типа данных — torch.float32, и цель set_default_dtype(torch.float64) — облегчить вывод типов, подобный NumPy. Значение по умолчанию для плавающего типа данных используется для:

1. Неявного определения значения по умолчанию для комплексного типа данных. Когда значение по умолчанию для плавающего типа — float32, значение по умолчанию для комплексного типа — complex64, а когда значение по умолчанию для плавающего типа — float64, значение по умолчанию для комплексного типа — complex128.
2. Вывода типа данных для тензоров, созданных с использованием чисел с плавающей точкой Python или комплексных чисел Python. См. примеры ниже.
3. Определения результата повышения типа между тензорами bool и integer и числами с плавающей точкой Python и комплексными числами Python.

Parameters

d (torch.dtype) — тип данных с плавающей точкой, который нужно сделать значением по умолчанию. Либо torch.float32, либо torch.float64.

Пример

>>> # initial default for floating point is torch.float32
>>> # Python floats are interpreted as float32
>>> torch.tensor([1.2, 3]).dtype
torch.float32
>>> # initial default for floating point is torch.complex64
>>> # Complex Python numbers are interpreted as complex64
>>> torch.tensor([1.2, 3j]).dtype
torch.complex64

>>> torch.set_default_dtype(torch.float64)

>>> # Python floats are now interpreted as float64
>>> torch.tensor([1.2, 3]).dtype    # a new floating point tensor
torch.float64
>>> # Complex Python numbers are now interpreted as complex128
>>> torch.tensor([1.2, 3j]).dtype   # a new complex tensor
torch.complex128