Deep-Learning-with-PyTorch

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位浮点数。

上一篇下一篇

猜你喜欢

热点阅读