类型信息

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

torch.finfo

class torch.finfo

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

一个 torch.finfo 提供以下属性

名称	类型	描述
bits	int	该类型占用的位数。
eps	float	最小的可表示数，满足 1.0 + eps != 1.0。
max	float	最大的可表示数。
min	float	最小的可表示数（通常为 -max）。
tiny	float	最小的正规数。等同于 smallest_normal。
smallest_normal	float	最小的正规数。参见注释。
resolution	float	该类型的近似十进制分辨率，即 10**-precision。

注意

torch.finfo 的构造函数可以在不带参数的情况下调用，此时该类将为 pytorch 默认 dtype（由 torch.get_default_dtype() 返回）创建。

注意

smallest_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 提供以下属性

名称	类型	描述
bits	int	该类型占用的位数。
max	int	最大的可表示数。
min	int	最小的可表示数。