类型信息

可以通过 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	最小的可表示数字。