torch:view和reshape的区别

复制代码
import torch

x = torch.randn(4, 6)      # shape: (4,6)
# view 用法
y = x.view(3, 8)           # shape: (3,8)
# reshape 用法
z = x.reshape(2, 12)       # shape: (2,12)
# 使用 -1 自动推断某一维度
w = x.view(-1, 8)          # 自动计算第一维为 3  (24/8=3)

注意:元素总数必须不变,否则抛出错误。

两者都用于改变张量的形状(不改变元素总数),但内部行为略有不同:

函数 连续性要求 内存行为 推荐场景
view 要求张量连续(contiguous) 返回视图,与原张量共享数据(零拷贝) 确定内存连续时使用,最高效
reshape 不要求连续 尽可能返回视图;若连续则等价于view,否则自动复制后再view 通用、安全,不用操心连续性
相关推荐
Easy_API4 小时前
2.8万亿参数,编程反超GPT-6 Astra:Kimi K3.1 Pro把开源模型推到了哪一步
大数据·人工智能·gpt·深度学习
代码山河4 小时前
Python就业方向全解析:后端、爬虫、数据分析、AI你选哪个?
python·学习·架构·教程·面向对象·项目
在世修行5 小时前
智学文枢:题目流水线
python·多线程·状态机·pyside
梯度下降者5 小时前
零基础入门Python 爬虫
爬虫·python·数据分析·入门指南·webcrawler
“AI国潮设计-小江”7 小时前
【AIGC实战】Python+Stable Diffusion打造“英歌舞财神”IP,解锁“马到成功”等爆款情绪营销文案设计
开发语言·人工智能·python·prompt·aigc
benchmark_cc8 小时前
用 Python 拉 A 股全市场实时快照:Notebook 请求、结果校验与易错点
python·数据分析·pandas·量化交易·股票数据·quantdash
伞伞悦读8 小时前
【第70期】Adam二阶矩不是方差:偏差校正、动量与AdamW完整实验
python
外收内放8 小时前
Python基础语法练习题(73)
java·前端·python
锋行天下9 小时前
Langgraph HITL 中断运行,用户输入后输出打字效果
人工智能·python
时早玖沐9 小时前
给 AstrBot 写一个天气推送插件:12 次提交、6 个坑
python·开源·ai编程