pytorch和numpy默认浮点类型位数详解

作者:gy笨瓜 时间:2021-11-10 10:19:49 

pytorch和numpy默认浮点类型位数

numpy中默认浮点类型为64位,pytorch中默认浮点类型位32位

测试代码如下

  • numpy版本:1.19.2

  • pytorch版本:1.2.0

In [1]: import torch
In [2]: import numpy as np
# 版本信息
In [3]: "pytorch version: {}, numpy version: {}".format(torch.__version__, np.__version__)
Out[3]: 'pytorch version: 1.2.0, numpy version: 1.19.2'

# numpy
In [4]: dat_np = np.array([1,2,3], dtype="float")
In [5]: dat_np.dtype
Out[5]: dtype('float64')

# pytorch
In [6]: dat_torch = torch.tensor([1,2,3])
In [7]: dat_torch = dat_torch.float()
In [8]: dat_torch.dtype
Out[8]: torch.float32

pytorch和numpy的默认类型与转换问题

pytorch对于浮点类型默认为float32,而numpy的默认类型是float64,转换的代码:

torch.from_numpy(a).type(torch.FloatTensor)
torch.from_numpy(np.float32(a))

来源:https://blog.csdn.net/u012633319/article/details/113784143

标签:pytorch,numpy,默认浮点类型,位数
0
投稿

猜你喜欢

  • Thinkphp5.0 框架使用模型Model添加、更新、删除数据操作详解

    2024-06-07 15:35:37
  • Python简单生成随机姓名的方法示例

    2023-04-26 19:35:58
  • ASP实现GB2312字符与区位码的相互转换

    2009-12-28 10:27:00
  • 如何从PyTorch中获取过程特征图实例详解

    2022-07-08 04:04:44
  • 教你使用Python pypinyin库实现汉字转拼音

    2021-10-03 16:48:36
  • Python文件的操作示例的详细讲解

    2023-11-20 17:54:55
  • Python实现邮件发送功能的示例详解

    2022-08-18 13:15:43
  • FrontPage2002简明教程二:文字与图像的处理

    2008-09-17 11:13:00
  • php+highchats生成动态统计图

    2024-05-02 17:19:08
  • 如何用变量实现群聊和悄悄话?

    2010-05-19 21:33:00
  • 通过两种方式增加从库——不停止mysql服务

    2024-01-17 17:44:28
  • Python xml、字典、json、类四种数据类型如何实现互相转换

    2023-10-17 17:36:20
  • tensorflow指定GPU与动态分配GPU memory设置

    2023-01-16 11:30:24
  • 怎么用FSO生成一个UNICODE格式的文本文件

    2009-05-26 15:40:00
  • php上传大文件设置方法

    2023-11-21 19:11:22
  • mysql8.0.30安装配置最详细教程(windows 64位)

    2024-01-16 10:01:35
  • 纯numpy卷积神经网络实现手写数字识别的实践

    2023-11-08 10:44:50
  • MySQL 8.0 新特性之检查约束的实现

    2024-01-23 15:25:08
  • IE bug: 1像素的dotted/dashed边框

    2009-10-28 18:45:00
  • MySQL存储引擎MyISAM与InnoDB区别总结整理

    2024-01-15 13:18:22
  • asp之家 网络编程 m.aspxhome.com