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) 等多用于不知道具体行数或列数时

相关推荐
敏叔V5871 分钟前
AI智能体的工具学习进阶:零样本API理解与调用
人工智能·学习
徐小夕@趣谈前端10 分钟前
拒绝重复造轮子?我们偏偏花365天,用Vue3写了款AI协同的Word编辑器
人工智能·编辑器·word
阿里云大数据AI技术11 分钟前
全模态、多引擎、一体化,阿里云DLF3.0构建Data+AI驱动的智能湖仓平台
人工智能·阿里云·云计算
鸽芷咕11 分钟前
DrissionPage 成 CANN 仓库爆款自动化工具:背后原因何在?
运维·python·自动化·cann
陈天伟教授11 分钟前
人工智能应用- 语言理解:05.大语言模型
人工智能·语言模型·自然语言处理
池央12 分钟前
CANN GE 深度解析:图编译器的核心优化策略、执行流调度与模型下沉技术原理
人工智能·ci/cd·自动化
爱学习的阿磊12 分钟前
使用Fabric自动化你的部署流程
jvm·数据库·python
七月稻草人15 分钟前
CANN ops-nn:AIGC底层神经网络算力的核心优化引擎
人工智能·神经网络·aigc·cann
种时光的人15 分钟前
CANN仓库核心解读:ops-nn打造AIGC模型的神经网络算子核心支撑
人工智能·神经网络·aigc
晚霞的不甘18 分钟前
守护智能边界:CANN 的 AI 安全机制深度解析
人工智能·安全·语言模型·自然语言处理·前端框架