pytorch和numpy默认浮点类型位数详解
更新时间：2023年02月02日 08:41:32 作者：gy笨瓜
这篇文章主要介绍了pytorch和numpy默认浮点类型位数，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教
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))
总结
以上为个人经验，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
这篇文章主要介绍了解决Python运算符重载的问题，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教2021-05-05
下面小编就为大家分享一篇Python数据分析库pandas基本操作方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2018-04-04
人工智能和区块链诞生至今已经有了十几年，当这些技术出现时，人们都说他们会改变世界，但至今为止，这两项技术对现实的影响依然有限，接下来通过本文介绍下从0编写区块链之用python解释区块链最基本原理,需要的朋友可以参考下2022-02-02
这篇文章主要介绍了浅谈python数据结构之动态规划,可能很多小伙伴会觉得这个词很陌生，觉得这是一种很复杂的思想，学习起来很困难，其实并不是这样，动态规划所讲述的知识与动态与规划并无太大关联,需要的朋友可以参考下2023-07-07
这篇文章主要介绍了Python捕捉和模拟鼠标事件的方法,涉及PyHook和PyWin32模块的使用技巧,需要的朋友可以参考下2015-06-06
有时在处理或打印一个PDF文档之前,你可能需要先知道该文档包含多少页,对于程序员来说,编写脚本来完成这项工作会更加高效,本文就介绍一个使用Python快速获取PDF文件页数的办法,需要的朋友可以参考下2024-03-03
这篇文章主要介绍了使用Python获取Linux系统的各种信息,例如系统类型、CPU信息、内存信息、块设备等,需要的朋友可以参考下2014-07-07
这篇文章主要介绍了解决Python 写文件报错TypeError的问题，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2020-10-10
详解Python 多线程 Timer定时器/延迟执行、Event事件
这篇文章主要介绍了Python 多线程 Timer定时器/延迟执行、Event事件的相关知识，本文给大家介绍的非常详细，具有一定的参考借鉴价值,需要的朋友可以参考下2019-06-06
这篇文章主要介绍了OpenCV基本操作之图像处理，本文给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下2021-09-09
最新评论