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中装饰器的理解与使用详解,装饰器本质上是一个闭包函数，其作用在于可以为其他函数增加额外功能，装饰器的返回值是一个函数对象,需要的朋友可以参考下2023-07-07
今天小编就为大家分享一篇python安装pywin32clipboard的操作方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2019-01-01
这篇文章主要给大家介绍了关于Django中CBV和FBV的相关资料，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧。2018-02-02
本文主要介绍了Django中外键ForeignKey介绍使用，文中通过示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2022-02-02
这篇文章主要介绍了在Python中实现字典反转案例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2020-12-12
这篇文章主要介绍了Python实现两个list求交集，并集，差集的方法,结合实例形式分析了Python使用intersection、union及difference方法实现两个集合list的交集、并集与差集操作技巧,需要的朋友可以参考下2018-08-08
Python使用selenium + headless chrome获取网页内容的方法示例
这篇文章主要介绍了Python使用selenium + headless chrome获取网页内容的方法示例，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2019-10-10
字符串是一种数据类型，但是，字符串比较特殊的是还有一个编码问题。下面这篇文章主要给大家介绍了关于Python3如何解决字符编码问题的相关资料，文中介绍的还是相对比较详细的，需要的朋友可以参考借鉴，下面来一起看看吧。2017-04-04
这篇文章主要介绍一个可以将音频提升音量的python常用工具，代码具有一定的学习价值，感兴趣的小伙伴可以了解一下2021-11-11
在日常工作中,我们经常需要给PDF文档添加一些标识,比如公司的图章或水印图章,所以本文就来为大家详细介绍一下如何使用Python实现为PDF文件添加图章,需要的可以参考下2023-11-11
最新评论