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

相关推荐
a里啊里啊1 分钟前
测试开发面试题
开发语言·chrome·python·xpath
豆沙糕3 分钟前
Python异步编程从入门到实战:结合RAG流式回答全解析
开发语言·python·面试
新缸中之脑22 分钟前
HDRI-Generator: 环境贴图生成AI
人工智能·贴图
网安情报局27 分钟前
企业押注Agentic SOC:AI重塑安全运营新范式
人工智能·网络安全
夜幕下的ACM之路29 分钟前
一、基础知识学习(Transformer + 上下文窗口 + Token 计算 + Embedding 向量)
人工智能·学习·transformer·embedding
东离与糖宝29 分钟前
LangChain4j vs Spring AI:最新对比,Java企业级Agent开发
java·人工智能
私人珍藏库33 分钟前
[Windows] 绘画工具 Krita v5.3.1
人工智能·windows·媒体·工具·软件·多功能
前端摸鱼匠36 分钟前
【AI大模型春招面试题13】残差连接(Residual Connection)与层归一化(Layer Norm)在Transformer中的作用?
人工智能·深度学习·语言模型·面试·transformer·求职招聘
乘凉~41 分钟前
【VideoCaptioner】开源音视频字幕自动识别工具
python