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

相关推荐
猿饵块27 分钟前
视觉slam--框架
人工智能
yvestine1 小时前
自然语言处理——Transformer
人工智能·深度学习·自然语言处理·transformer
SuperW2 小时前
OPENCV图形计算面积、弧长API讲解(1)
人工智能·opencv·计算机视觉
山海不说话2 小时前
视频行为标注工具BehaviLabel(源码+使用介绍+Windows.Exe版本)
人工智能·python·计算机视觉·视觉检测
虹科数字化与AR3 小时前
安宝特方案丨船舶智造的“AR+AI+作业标准化管理解决方案”(装配)
人工智能·ar·ar眼镜·船舶智造·数字工作流·智能装配
liuzhenghua664 小时前
Python任务调度模型
java·运维·python
飞哥数智坊5 小时前
Coze实战第13讲:飞书多维表格读取+豆包生图模型,轻松批量生成短剧封面
人工智能
小前端大牛马5 小时前
java教程笔记(十一)-泛型
java·笔记·python
sjtu_cjs5 小时前
Tensorrt python api 10.11.0笔记
开发语言·笔记·python
哆啦A梦的口袋呀5 小时前
深入理解系统:UML类图
开发语言·python·uml