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)

相关推荐
Peter·Pan爱编程几秒前
10. new_delete 不是 malloc_free 的包装
c++·人工智能·算法
IT_陈寒6 分钟前
Vue的computed属性怎么突然不更新了?
前端·人工智能·后端
方向研究15 分钟前
人类的核心能力
人工智能
测试员周周21 分钟前
【Appium 系列】第18节-重试与容错 — 移动端测试的稳定性保障
人工智能·python·功能测试·ui·单元测试·appium·测试用例
灰灰勇闯IT41 分钟前
DeepSeek-R1 在 CANN 上的推理部署
pytorch·python·深度学习
l1t43 分钟前
Hy-MT2-1.8B总结的pgvector 0.8.2解决了并行HNSW索引构建漏洞
数据库·人工智能·postgresql
太华43 分钟前
学习AI Agent编程-第二天-LangGraph ReAct模式实现
人工智能
dayuOK63071 小时前
从“爆款复刻”到“个性化创作”:AI辅助写作的技术挑战与演进方向
人工智能·职场和发展·自动化·新媒体运营·媒体
努力学习_小白1 小时前
Inception V1——学习记录
pytorch·深度学习·inception v1
Raink老师1 小时前
【AI面试临阵磨枪-58】AI 生成内容合规、版权、审核机制设计
人工智能·面试·职场和发展