x.view()

x.view()就是对tensor进行reshape

python 复制代码
import torch
v1 = torch.range(1, 4)
v2 = v1.view(2, 2)
print(v2)
v3 = v2.view(4,-1)
print(v3)
python 复制代码
tensor([[1., 2.],
        [3., 4.]])
tensor([[1.],
        [2.],
        [3.],
        [4.]])

在函数的参数中经常可以看到-1例如x.view(-1, 4)

这里-1表示一个不确定的数,就是你如果不确定你想要reshape成几行,但是你很肯定要reshape成4列,那不确定的地方就可以写成-1

例如一个长度的16向量x,

x.view(-1, 4)等价于x.view(4, 4)

x.view(-1, 2)等价于x.view(8,2)

相关推荐
Dev7z3 分钟前
基于深度学习的口腔疾病自主诊断系统设计与实现(UI界面+训练代码+数据集)
人工智能·深度学习·口腔疾病
IT大师兄吖3 分钟前
whisperx 语音转字幕 懒人整合包
人工智能
前端不太难4 分钟前
AI 驱动游戏:鸿蒙生态的机会在哪里?
人工智能·游戏·harmonyos
鹧鸪云光伏5 分钟前
AI赋能储能设计系统,让设计更简单
人工智能·信息可视化·数据分析·光伏·储能设计方案
fof9205 分钟前
Base LLM | 从 NLP 到 LLM 的算法全栈教程 第五天
人工智能·自然语言处理
乘凉~6 分钟前
【MoneyPrinterTurbo】一个利用AI大模型一键生成高清短视频的工具,开源免费,只要有电脑就能用
人工智能·电脑·音视频
专精特新小巨人10 分钟前
技术突破!百创智造FFPE空间转录组新品问世,攻克高度降解样本检测难题
人工智能
模拟器连接器曾工10 分钟前
AI视觉检测设备参数有哪些?从硬件到算法的全面解析
人工智能·算法·视觉检测·ai视觉·ai视觉检测
CV-deeplearning11 分钟前
Claude Code 深度体验:Anthropic 官方 AI 编程助手,让终端变成你的超级搭档!
人工智能·claude code
量子物理学11 分钟前
Open CV 边缘检测算法:Canny、Sobel、Scharr与Laplacian对比解析
人工智能·算法·计算机视觉