Shortcuts

torch.get_default_dtype

torch.get_default_dtype() torch.dtype

获取当前默认的浮点数 torch.dtype

示例:

>>> torch.get_default_dtype()  # 初始默认浮点类型为 torch.float32
torch.float32
>>> torch.set_default_dtype(torch.float64)
>>> torch.get_default_dtype()  # 默认值现已更改为 torch.float64
torch.float64
优云智算