脚本之家 服务器常用软件
快捷导航
您的位置：首页 → 脚本专栏 → python → pytorch Tensor类型

pytorch常见的Tensor类型详解

 更新时间：2020年01月15日 10:45:59   作者：fly_Xiaoma  
今天小编就为大家分享一篇pytorch常见的Tensor类型详解，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧

Tensor有不同的数据类型，每种类型分别有对应CPU和GPU版本（HalfTensor除外）。默认的Tensor是FloatTensor，可通过torch.set_default_tensor_type修改默认tensor类型（如果默认类型为GPU tensor，则所有操作都将在GPU上进行）。

Tensor的类型对分析内存占用很有帮助，例如，一个size为(1000,1000,1000)的FloatTensor，它有1000*1000*1000=10^9个元素，每一个元素占用32bit/8=4Byte内存，所以共占用大约4GB内存/显存。HalfTensor是专为GPU版本设计的，同样的元素个数，显存占用只有HalfTensor的一半，所以可以极大缓解GPU显存不足的问题，但是由于HalfTensor所能表示的数值大小和精度有限，所以可能出现溢出等问题。

数据类型 CPU Tensor GPU Tensor
32 bit 浮点 torch.FloatTensor torch.cuda.FloatTensor
64 bit 浮点 torch.DoubleTensor torch.cuda.DoubleTensor
16 bit 半精度浮点 N/A torch.cuda.HalfTensor
8 bit 无符号整形(0~255) torch.ByteTensor torch.cuda.ByteTensor
8 bit 有符号整形(-128~127) torch.CharTensor torch.cuda.CharTensor
16 bit 有符号整形 torch.ShortTensor torch.cuda.ShortTensor
32 bit 有符号整形 torch.IntTensor torch.cuda.IntTensor
64 bit 有符号整形 torch.LongTensor torch.cuda LongTensor

各数据类型之间可以互相转换，type(new_type)是通用的做法，同时还有float、long、half等快捷方法。CPU tensor和GPU tensor之间的互换是通过tensor.cuda和tensor.cpu的方法实现。

如：

#设置默认tensor，注意参数是字符串
torch.set_default_tensor_type('torch.IntTensor')
 
a=torch.Tensor(2,3)
print(a)  #a现在是IntTensor

以上这篇pytorch常见的Tensor类型详解就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。

您可能感兴趣的文章:

相关文章

  • Python实现购物系统(示例讲解)

    Python实现购物系统(示例讲解)

    下面小编就为大家带来一篇Python实现购物系统(示例讲解)。小编觉得挺不错的，现在就分享给大家，也给大家做个参考。一起跟随小编过来看看吧
    2017-09-09
  • Python常驻任务实现接收外界参数代码解析

    Python常驻任务实现接收外界参数代码解析

    这篇文章主要介绍了Python常驻任务实现接收外界参数代码解析,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
    2020-07-07
  • Python打包文件执行报错：ModuleNotFoundError: No module named ‘pymssql‘的解决方法

    Python打包文件执行报错：ModuleNotFoundError: No module 

    这篇文章给大家介绍了Python打包文件执行报错：ModuleNotFoundError: No module named ‘pymssql‘的解决方法,如果有遇到相同问题的朋友可以参考阅读一下本文
    2023-10-10
  • YOLOv5改进教程之添加注意力机制

    YOLOv5改进教程之添加注意力机制

    注意力机制最先被用在NLP领域,Attention就是为了让模型认识到数据中哪一部分是最重要的,为它分配更大的权重,获得更多的注意力在一些特征上,让模型表现更好,这篇文章主要给大家介绍了关于YOLOv5改进教程之添加注意力机制的相关资料,需要的朋友可以参考下
    2022-06-06
  • python填充彩色图形的实现示例

    python填充彩色图形的实现示例

    本文主要介绍了python填充彩色图形的实现示例，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
    2023-04-04
  • python光学仿真相速度和群速度计算理解学习

    python光学仿真相速度和群速度计算理解学习

    从物理学的机制出发，波动模型相对于光线模型，显然更加接近光的本质；但是从物理学的发展来说，波动光学旨在解决几何光学无法解决的问题，可谓光线模型的一种升级
    2021-10-10
  • Pycharm IDE的安装和使用教程详解

    Pycharm IDE的安装和使用教程详解

    这篇文章主要介绍了Pycharm IDE的安装和使用,本文通过图文并茂的形式给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下
    2020-04-04
  • Pandas数据分析之pandas文本处理

    Pandas数据分析之pandas文本处理

    这篇文章主要介绍了Pandas数据分析之pandas文本处理，pandas对文本数据也有很多便捷处理方法，可以不用写循环，向量化操作运算速度快，还可以进行高级的正则表达式，各种复杂的逻辑筛选和匹配提取信息
    2022-08-08
  • PyQt5下拉式复选框QComboCheckBox的实例

    PyQt5下拉式复选框QComboCheckBox的实例

    今天小编就为大家分享一篇PyQt5下拉式复选框QComboCheckBox的实例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
    2019-06-06
  • python中正则的使用指南

    python中正则的使用指南

    正则表达式（RE）是一种小型的、高度专业化的编程语言，它内嵌在Python中，并通过re模块实现。下面我们就来详细探讨下Python中正则表达式的使用
    2016-12-12

最新评论