7.第七个方法
torch.get_default_dtype()
- 如果你看过我的上一个方法
torch.set_defautl_dtype())
,那么这个方法马上就能理解,这两个方法是一对的,torch.set_defautl_dtype())
设置默认浮点类型,而torch.get_defautl_dtype())
获取默认浮点类型。 - 在pytorch中,我们的浮点tensor如果我们不指定其数据类型的话系统会给它一个默认的类型,此方法的作用就是获得这个类型。
import torch
a = torch.tensor([1.,2.])
a.dtype
- 可以知道,这个默认类型是
torch.float32
,我们调用此方法
- 所以此方法返回的就是默认的float类型,如果我们改变默认类型
- 那么对应返回的类型也就改变了。