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]

相关推荐
Tezign_space8 小时前
AI用户洞察新纪元:atypica.AI如何重塑商业决策逻辑
人工智能·ai智能体·atypica
却道天凉_好个秋8 小时前
OpenCV(十一):色彩空间转换
人工智能·opencv·计算机视觉
跳跳的向阳花8 小时前
01、大模型部署方案与Dify的使用
python
ezl1fe8 小时前
RAG 每日一技(二十一):让证据“会算账”——差异对照与可核验公式的最小闭环
人工智能·后端·程序员
电鱼智能的电小鱼8 小时前
服装制造企业痛点解决方案:EFISH-SBC-RK3588 预测性维护方案
网络·人工智能·嵌入式硬件·算法·制造
用户5191495848458 小时前
云防护栏理论:应对云配置错误的安全防护策略
人工智能·aigc
_Mya_8 小时前
后端接口又改了?让 Apifox MCP 帮你自动同步类型定义
前端·人工智能·mcp
西柚小萌新8 小时前
【Python从入门到精通】--Pycharm增加内存
开发语言·python·pycharm
西柚小萌新8 小时前
【深入浅出PyTorch】--7.1.PyTorch可视化1
人工智能·pytorch·python
我是华为OD~HR~栗栗呀9 小时前
华为OD-23届考研-Java面经
java·c++·后端·python·华为od·华为·面试