Deep-Learning-with-PyTorch-3.5.1 用dtype指定数字类型

3.5.1 用dtype指定数字类型

张量构造函数的dtype参数(即像张量,零和一的函数)指定将包含在张量中的数值数据(d)类型。 数据类型指定张量可以容纳的可能值(整数与浮点数)以及每个值的字节数。 dtype参数故意与同名的标准NumPy参数相似。 这是dtype参数的可能值的列表:

torch.float32或torch.float:32位浮点

torch.float64或torch.double:64位,双精度浮点

torch.float16或torch.half:16位半精度浮点

torch.int8:带符号的8位整数

torch.uint8:无符号的8位整数

torch.int16或torch.short:带符号的16位整数

torch.int32或torch.int:带符号的32位整数

torch.int64或torch.long:带符号的64位整数

torch.bool:布尔值

张量的默认数据类型是32位浮点数。

©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。