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)

相关推荐
qzhqbb12 小时前
论文精读:GrammarGPT——基于开源大模型与混合数据的中文母语语法纠错模型
人工智能
EnCi Zheng13 小时前
09ba-斯坦福CS336作业一-前馈网络
人工智能·transformer
大鹏的NLP博客13 小时前
类别不平衡与加权交叉熵
人工智能·机器学习·图像检测
Mr.朱鹏13 小时前
科技资讯日报 · 2026-06-15
人工智能·科技·ai·chatgpt
逻辑君13 小时前
认知神经科学研究报告【20260089】
人工智能·深度学习·机器学习
WangN213 小时前
【通识】宇树G1_29DOF速度跟踪训练—逐章学习手册
人工智能·python·学习·机器人·具身智能
nbtang202613 小时前
每日AI新闻推送 | 2026年6月12日
人工智能
邵宇然13 小时前
轻量级推理引擎开发:从模型加载到推理执行的 Rust 实战
人工智能
装不满的克莱因瓶13 小时前
掌握语义分割经典模型 FCN——从像素分类到端到端分割的奠基之作
人工智能·python·深度学习·算法·机器学习·分类·数据挖掘
ACP广源盛1392462567313 小时前
GSV5600@ACP#多接口协议转换芯片,物理 AI 便携终端的互联核心
大数据·人工智能·分布式·嵌入式硬件·spark