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类型,如果我们改变默认类型

- 那么对应返回的类型也就改变了。
本文介绍了PyTorch中获取默认浮点数类型的方法torch.get_default_dtype()。此方法用于查询系统为未指定数据类型的浮点张量所设定的默认类型。通过示例展示如何使用该方法,并解释了其与torch.set_default_dtype()方法的关系。


被折叠的 条评论
为什么被折叠?



