pytorch算子torch.arange在CPU GPU NPU中支持数据类型格式
更新时间：2022年09月19日 16:22:52 作者：ctrlA_ctrlC_ctrlV
这篇文章主要为大家介绍了pytorch算子torch.arange在CPU GPU NPU支持数据类型格式，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪
正文
CPU（Central Processing Unit）：中央处理器 GPU（Graphics Processing Unit）：图形处理器 NPU（Neural Network Processing Unit）：神经网络处理器，是基于神经网络算法与加速的新型处理器总称。
一、 torch.arange() 和 torch.range() 的用法
pytorch官网介绍：
torch.arange(start,end,step) 用于产生一个从start开始,到end结束（注意不包括end），步长为step的Tensor, 并且可以设置 Tensor 的 device 和 dtype
torch.arange 与 torch.range 功能及其相似，不同之处在于 torch.range(start,end,step) 生成的 Tensor, 包括 end
如：
a=torch.arange(1, 7, 2) b=torch.range(1, 7, 2) print(a) print(b)
输出：
tensor([1, 3, 5]) tensor([1., 3., 5., 7.])
但是建议使用 torch.arange ，因为 torch.range 即将被pytorch 移除：
二、 torch.arange 支持的数据类型格式
只考虑 float 类型
cpu 不支持 float16，支持 float32 和 float64 cpu 支持 float16 、float32 和 float64 npu 不支持 float16 和 float64 ，只支持 float32
事实上 npu 基本不支持所有的 64位类型，包括 int64 和 float64，与算子无关。当然cpu 是支持 16位数据类型的，只是 torch.arange 不支持而已。
验证代码如下：
import torch # CPU a=torch.arange(1, 10, 2,device="cpu",dtype=torch.float16) # 不可以 a=torch.arange(1, 10, 2,device="cpu",dtype=torch.float32) # 可以 a=torch.arange(1, 10, 2,device="cpu",dtype=torch.float64) # 可以 # GPU a=torch.arange(1, 10, 2, device="cuda:0",dtype=torch.float16) # 可以 a=torch.arange(1, 10, 2, device="cuda:0",dtype=torch.float32) # 可以 a=torch.arange(1, 10, 2, device="cuda:0",dtype=torch.float64) # 可以 # NPU a=torch.arange(1, 10, 2, device="npu:0",dtype=torch.float16) # 不可以 a=torch.arange(1, 10, 2, device="npu:0",dtype=torch.float32) # 可以 a=torch.arange(1, 10, 2, device="npu:0",dtype=torch.float64) # 不可以 print(a)
以上就是pytorch算子torch.arange在CPU GPU NPU中支持数据类型格式的详细内容，更多关于pytorch算子支持数据类型格式的资料请关注脚本之家其它相关文章！
相关文章
Python ORM框架SQLAlchemy学习笔记之数据添加和事务回滚介绍
这篇文章主要介绍了Python ORM框架SQLAlchemy学习笔记之数据添加和事务回滚介绍,需要的朋友可以参考下2014-06-06
这篇文章主要为大家详细介绍了python实现AES加密与解密，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2019-03-03
这篇文章主要介绍了Python脚本实现集群检测和管理功能,本文讲解了实现想法、开发工具选择、经验分享、代码示例等内容,需要的朋友可以参考下2015-03-03
工作上经常需要与外国友人邮件沟通，奈何工作电脑没有安装有道词典一类的翻译软件，结合自己的需要，自己用PySimpleGUI撸一个桌面翻译神器，感兴趣的可以了解一下2022-09-09
这篇文章主要介绍了python爬虫模块URL管理器模块用法解析,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2020-02-02
这篇文章主要介绍了手把手教你安装Windows版本的Tensorflow，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2020-03-03
这篇文章主要介绍了Python处理mat文件的三种方式小结，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教2022-05-05
Python Allure库是一个实用可靠的测试报告框架,它几乎可以与Python的其他库和框架无缝集成,利用Python Allure库,可以轻松生成易于阅读的测试报告,让测试变得更加简单便捷,本文主要介绍Python Allure库的使用,感兴趣的朋友一起看看吧2023-12-12
这篇文章主要介绍了python实现360的字符显示界面示例,需要的朋友可以参考下2014-02-02
这篇文章主要为大家详细介绍了Python echarts实现数据可视化，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下，希望能够给你带来帮助2022-03-03
最新评论