类型信息

可以通过 torch.finfo 或 torch.iinfo 访问 torch.dtype 的数值属性。

torch.finfo

class torch.finfo

一个 torch.finfo 是一个表示浮点数 torch.dtype 的数值属性的对象，（即 torch.float32、torch.float64、torch.float16 和 torch.bfloat16）。这类似于 numpy.finfo。

一个 torch.finfo 提供了以下属性：

名称	类型	描述
位	整数	该类型占用的位数。
eps	浮点数	最小的可表示数字，使得 1.0 + eps != 1.0。
最大值	浮点数	可表示的最大数值。
最小值	浮点数	可表示的最小数字（通常是 -max）。
微小	浮点数	最小的正正规数。等价于 smallest_normal。
最小正规值	浮点数	最小的正正规数。参见注释。
分辨率	浮点数	这种类型的近似十进制分辨率，即，10**-precision。

注意

构造函数 torch.finfo 可以在没有参数的情况下调用，在这种情况下，类是为 pytorch 默认 dtype 创建的（如 torch.get_default_dtype() 返回的）。

注意

smallest_normal 返回最小的 normal 数，但存在更小的 次正规数。更多信息请参见 https://en.wikipedia.org/wiki/Denormal_number。

torch.iinfo

class torch.iinfo

一个 torch.iinfo 是一个表示整数 torch.dtype 数值属性的对象（即 torch.uint8、torch.int8、torch.int16、torch.int32 和 torch.int64）。这类似于 numpy.iinfo。

一个 torch.iinfo 提供了以下属性：

名称	类型	描述
位	整数	该类型占用的位数。
最大值	整数	可表示的最大数值。
最小值	整数	可表示的最小数字。