torch.set_default_tensor_type

torch.set_default_tensor_type(t, /)

警告

此函数自 PyTorch 2.1 起已弃用，请改用 torch.set_default_dtype() 和 torch.set_default_device()。

将默认的 torch.Tensor 类型设置为浮点张量类型 t。此类型也将用作 torch.tensor() 中类型推断的默认浮点类型。

默认的浮点张量类型最初为 torch.FloatTensor。

参数

t (type or string) – 浮点张量类型或其名称

示例

>>> torch.tensor([1.2, 3]).dtype    # initial default for floating point is torch.float32
torch.float32
>>> torch.set_default_tensor_type(torch.DoubleTensor)
>>> torch.tensor([1.2, 3]).dtype    # a new floating point tensor
torch.float64