pytorch基本数据类型

目录

  • [1. tensor](#1. tensor)
  • [2. 数据类型](#2. 数据类型)
  • [3. 数据类型的创建和使用](#3. 数据类型的创建和使用)
    • [3.1 数据类型check](#3.1 数据类型check)
    • [3.2 Dimension为0的tensor](#3.2 Dimension为0的tensor)
    • [3.2 Dimension为1的tensor](#3.2 Dimension为1的tensor)
    • [3.3 Dimension为2的tensor](#3.3 Dimension为2的tensor)
    • [3.4 Dimension为3的tensor](#3.4 Dimension为3的tensor)
    • [3.5 Dimension为4的tensor](#3.5 Dimension为4的tensor)

1. tensor

下图为python数据类型与pytorch的对应关系

在pytorch中是没有string类型的,如果要使用string类型,需要借助两种手段:

2. 数据类型

从图中可以看出pytorch数据类型是分cpu和GPU的,gpu是在torch后加一个cuda,他们是完全不同的数据

这里经常使用的数据类型是FloatTensor和ByteTensor

3. 数据类型的创建和使用

3.1 数据类型check


torch.randn是生成随机初始化的2行3列的tensor数据

python 复制代码
a = torch.randn(2, 3)
print(a)
tensor([[ 0.4406,  0.4335, -1.6466],
        [-0.1927,  0.4834,  1.0126]])

3.2 Dimension为0的tensor

Dimension为0的tensor多用来表示loss的值

.shape是个成员,返回torch.Size([])表示0维,.size()是成员函数返回结果与.shape一样

3.2 Dimension为1的tensor

一维tensor多用于bias和线性数据输入,如下图:

3.3 Dimension为2的tensor

当线性输入,一次输入多张图片时使用2维tensor,比如:使用minist数据集做手写数字识别,表示为[n, 784] n表示一次输入几张图片。

3.4 Dimension为3的tensor

3维多用于RNN,NLP文字处理,[10,20,100] 表示20个句子每个句子10个单词,单词用100向量表示。

3.5 Dimension为4的tensor

4维多用于CNN卷积循环神经网络,[b,c,h,w]分别表示为[batch, channel, height, weight]

相关推荐
koo3646 小时前
李宏毅机器学习笔记30
人工智能·笔记·机器学习
长桥夜波7 小时前
机器学习日报02
人工智能·机器学习·neo4j
Elastic 中国社区官方博客7 小时前
介绍 Elastic 的 Agent Builder - 9.2
大数据·运维·人工智能·elasticsearch·搜索引擎·ai·全文检索
拓端研究室7 小时前
专题:2025年制造业数智化发展白皮书:数字化转型与智能制造|附130+份报告PDF、数据、绘图模板汇总下载
人工智能
就不爱吃大米饭7 小时前
ChatGPT官方AI浏览器正式推出:ChatGPT Atlas浏览器功能及操作全解!
人工智能·chatgpt
卡卡恩7 小时前
使用uv创建系统全局python执行环境
python
牛客企业服务7 小时前
企业招聘新趋势:「AI面试」如何破解在线作弊难题?
人工智能·面试·职场和发展·招聘·ai招聘
查士丁尼·绵7 小时前
笔试-座位调整
python
infominer7 小时前
数据处理像搭乐高?详解 RAGFlow Ingestion Pipeline
人工智能·ai-native
飞翔的佩奇7 小时前
【完整源码+数据集+部署教程】【运动的&足球】足球场地区域图像分割系统源码&数据集全套:改进yolo11-RFAConv
前端·python·yolo·计算机视觉·数据集·yolo11·足球场地区域图像分割系统