美文网首页Deep-Learning-with-PyTorch
Deep-Learning-with-PyTorch-3.5.1

Deep-Learning-with-PyTorch-3.5.1

作者: 追求科技的足球 | 来源:发表于2020-09-28 20:11 被阅读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位浮点数。

    相关文章

      网友评论

        本文标题:Deep-Learning-with-PyTorch-3.5.1

        本文链接:https://www.haomeiwen.com/subject/zdjjuktx.html