数据类型信息

优质
小牛编辑
128浏览
2023-12-01

译者:冯宝宝

可以通过torch.finfotorch.iinfo访问torch.dtype的数字属性。

torch.finfo

class torch.finfo

torch.finfo 是一个用来表示浮点torch.dtype的数字属性的对象(即torch.float32torch.float64torch.float16)。 这类似于 numpy.finfo。

torch.finfo 提供以下属性:

名称类型描述
bits整型 int数据类型占用的位数
eps浮点型float可表示的最小数字,使得1.0 + eps!= 1.0
max浮点型float可表示的最大数字
tiny浮点型float可表示的最小正数

注意

在使用pytorch默认dtype创建类(由torch.get_default_dtype()返回)的情况下,构造的 torch.finfo 函数可以不带参数被调用。

torch.iinfo

class torch.iinfo

torch.iinfo是一个用来表示整数torch.dtype 的数字属性的对象,(即torch.uint8torch.int8torch.int16torch.int32torch.int64)。 这与numpy.iinfo类似。

torch.iinfo 提供以下属性:

名称类型描述
bits整型数据类型占用的位数
max整型可表示的最大数字