Deep-Learning-with-PyTorch-3.5.1
2020-09-28 本文已影响0人
追求科技的足球
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位浮点数。