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)

相关推荐
猫头虎12 小时前
FDE 是什么岗位?Forward Deployed Engineer 岗位标准、能力模型、交付物规范与冲突处置规则全解析
人工智能·开源·开源软件·ai编程·ai写作·gpu算力·agi
G311354227312 小时前
当声音、图片和视频都能被生成,信任将从“看起来真实”转向“能够验证”
人工智能
揽秀亭长12 小时前
视频转脚本实际怎么做?对比3个不同方案,拆解视频转脚本不同技术流程
人工智能·音视频
知几蜗牛12 小时前
GPU抢不到就换一种:训练任务需要先声明可替代性
人工智能
知几蜗牛12 小时前
Agent不是多想几步就能上线:用状态机管住自动行动
人工智能
adinnet202612 小时前
客服与工单:响应时长与满意度问数
数据库·人工智能
知几蜗牛12 小时前
数据不能集中,算力也不统一:联邦学习终于面对运维现实
人工智能
知几蜗牛12 小时前
OpenAI开始公开模型失配个案:真正重要的是这套报告制度
人工智能
知几蜗牛13 小时前
广告开始和你对话:Sponsored Agents改变的不是文案
人工智能
ADAI_Bowen13 小时前
聚焦集成与精度:建筑室内 AI 平台实测解析,本土方案与渲染工具对比
人工智能