torch.set_default_tensor_type

torch.set_default_tensor_type(t, /)[source][source]

警告

此函数自 PyTorch 2.1 版本开始已弃用，请使用 torch.set_default_dtype() 和 torch.set_default_device() 作为替代。

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

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

参数:

t（类型或字符串）- 浮点张量类型或其名称

示例：

>>> 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