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

相关推荐
新缸中之脑几秒前
RAG 陷阱:向量搜索不是语义理解
人工智能
浔川python社2 分钟前
浔川社团关于产品数据情况的官方通告
python
生活很暖很治愈4 分钟前
GUI自动化测试[3]——控件&数鼠标操作
windows·python·功能测试·测试工具
EmmaXLZHONG5 分钟前
Reinforce Learning Concept Flow Chart (强化学习概念流程图)
人工智能·深度学习·机器学习·流程图
薛定谔的猫19827 分钟前
十三.调用 BERT 中文文本情感分析交互式推理模型训练好的
人工智能·深度学习·bert
home_4989 分钟前
与gemini关于宇宙观科幻对话
人工智能
Candice Can10 分钟前
【机器学习】吴恩达机器学习Lecture2-Linear regression with one variable
人工智能·机器学习·线性回归·吴恩达机器学习
undsky_15 分钟前
【RuoYi-SpringBoot3-Pro】:将 AI 编程融入传统 java 开发
java·人工智能·spring boot·ai·ai编程
老蒋每日coding15 分钟前
Python3基础练习题详解,从入门到熟练的 50 个实例(一)
开发语言·python
薛定谔的猫198217 分钟前
十二、基于 BERT 的中文文本二分类模型测试实战:从数据加载到准确率评估
人工智能·分类·bert