脚本之家 服务器常用软件
快捷导航
您的位置：首页 → 脚本专栏 → python → pytorch算子支持数据类型格式

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先序遍历二叉树问题

    python先序遍历二叉树问题

    这篇文章主要介绍了python先序遍历二叉树问题，简单分析了问题，然后向大家分享了代码示例，具有一定参考价值，需要的朋友可以了解下。
    2017-11-11
  • Python就将所有的英文单词首字母变成大写

    Python就将所有的英文单词首字母变成大写

    这篇文章主要给大家介绍了关于Python就将所有的英文单词首字母变成大写的相关资料，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
    2021-02-02
  • 解决win64 Python下安装PIL出错问题(图解)

    解决win64 Python下安装PIL出错问题(图解)

    这篇文章主要介绍了解决win64 Python下安装PIL出错问题,文中的解决方法也很简单，需要的朋友参考下吧
    2018-09-09
  • 教你用Python实现自动提取并收集信息的功能

    教你用Python实现自动提取并收集信息的功能

    今天教大家怎么用Python实现自动提取并收集信息的功能,文中介绍的非常详细,有很多代码示例,对正在学习python的小伙伴们很有帮助,需要的朋友可以参考下
    2021-05-05
  • 浅谈Python3中strip()、lstrip()、rstrip()用法详解

    浅谈Python3中strip()、lstrip()、rstrip()用法详解

    这篇文章主要介绍了浅谈Python3中strip()、lstrip()、rstrip()用法详解，小编觉得挺不错的，现在分享给大家，也给大家做个参考。一起跟随小编过来看看吧
    2019-04-04
  • python3实现UDP协议的服务器和客户端

    python3实现UDP协议的服务器和客户端

    这篇文章主要为大家详细介绍了python3实现UDP协议的服务器和客户端，具有一定的参考价值，感兴趣的小伙伴们可以参考一下
    2017-06-06
  • python执行get提交的方法

    python执行get提交的方法

    这篇文章主要介绍了python执行get提交的方法,实例分析了Python通过get传递数据的技巧,具有一定参考借鉴价值,需要的朋友可以参考下
    2015-04-04
  • Pygame transform模块入门介绍

    Pygame transform模块入门介绍

    pygame.transform模块允许您对加载、创建后的图像进行一系列操作，比如调整图像大小、旋转图片等操作，感兴趣的可以了解一下
    2022-08-08
  • python3 gunicorn配置文件的用法解读

    python3 gunicorn配置文件的用法解读

    这篇文章主要介绍了python3 gunicorn配置文件的使用,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教
    2025-04-04
  • 浅析Python中的缩进错误

    浅析Python中的缩进错误

    在编程中,我们经常会遇到错误,缩进错误是 Python 中最常见的错误之一,它会使我们的代码难以理解,并且难以调试,下面小编就来和大家简单聊聊Python中的缩进错误吧
    2023-10-10

最新评论