Pytorch reshape用法

这里-1是指未设定行数,程序自动计算,所以这里-1表示任一正整数

example

reshape(-1, 1) 表示(任意行,1列),4行4列变为16行1列
reshape(1, -1) 表示(1行,任意列),4行4列变为1行16列
reshape(-1, 2) 表示(任意行,2列),4行4列变为8行2列

reshape(-1, 1) 等多用于不知道具体行数或列数时

相关推荐
惊鸿一博11 小时前
基于Diffusion的轨迹优化:扩散模型在自动驾驶中的原理、架构与落地挑战
人工智能·自动驾驶·diffusion·规控
合合技术团队11 小时前
合合信息携手中科软发布智能化方案,推动AI融入保险理赔、核保与风控多节点
大数据·人工智能·文档智能·数据智能
zyplayer-doc11 小时前
核心制度不该被随手修改:用zyplayer-doc锁定文档和全部子文档
大数据·数据库·人工智能·笔记·pdf·ocr
具身新纪元11 小时前
CVPR 2026|新缺陷不断上线,如何让工业视觉检测模型不忘旧账?
人工智能·深度学习·目标检测·机器学习·计算机视觉·目标跟踪·视觉检测
Raas10011 小时前
MAIGateway,魔芋企业级AI网关的FinAPI成本弹性设计
人工智能
weixin_BYSJ198711 小时前
【java项目分享】springboot阅读推荐平台10600
java·javascript·spring boot·python·django·flask·php
constCpp11 小时前
AI 编程:追不完的工具,理得清的问题
人工智能·ai编程·ai-native
2401_8652616311 小时前
亦唐科技(YIKTONG):驱动国产贴片机迈向智能化未来
大数据·人工智能
阿kun要赚马内11 小时前
工具在langchain agent中的调用
人工智能·后端·python
IT_陈寒11 小时前
Vite的HMR在我项目上突然失效,排查三天找到离谱原因
前端·人工智能·后端