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,默认浮点类型,位数


猜你喜欢
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
