类型信息

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 默认数据类型创建（如 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	最小的可表示数。