当前位置: 首页 > news >正文

每天五分钟深度学习框架pytorch:多维tensor向量在某一维度的拼接和分割

本文重点

在深度学习中,我们常常需要完成多个向量拼接,同时也要完成向量的分割,在pytorch中已经有封装好的库,我们可以直接调用完成这部分任务。

Cat拼接

c=torch.cat([a,b],dim=0)表示将a和b按0维度进行拼接,需要注意再非dim维度,两个矩阵的维度必须是一致的,不然会拼接失败。

Stack拼接

Stack并不会进行维度的拼接,而是会增加新的维度

我们可以看到将a和b在dim=0的维度进行stack,那么并不会将dim=0的16加起来组成32,3,28,28,而是会增加一个新的维度2,这是与cat不一样的地方。

注意stack的两个列表必须在所有维度上都一样大小,如果不一样大小则会出错

Split切割

http://www.lryc.cn/news/374335.html

相关文章:

  • 从C语言到C++(五)
  • 数据结构——栈(Stack)详解
  • 1.Element的table表高度自适应vue3+js写法
  • 联想电脑电池只能充到80%,就不在充电了,猛一看以为坏了,只是设置了养护模式。
  • Unity接入PS5手柄和Xbox手柄以及Android平台的(以及不同平台分析)
  • vue+java实现简易AI问答组件(基于百度文心大模型)
  • 刷代码随想有感(104):动态规划——01背包问题/二维dp数组
  • Docker-Portainer可视化管理工具
  • SqlSugar 集成
  • MySQL Connector/C++ 和 MySQL Connector/ODBC 的区别
  • Weevil-Optimizer象鼻虫优化算法的matlab仿真实现
  • Web前端项目-交互式3D魔方【附源码】
  • 视频格式转换avi格式怎么弄?分享视频转换方法
  • UniRx 入门
  • 简单游戏制作——飞行棋
  • 等保一体机
  • 什么是寄存器文件(Register File)?
  • 6月15号作业
  • 零基础入门学用Arduino 第三部分(三)
  • Trusty qemu + android环境搭建详细步骤
  • 杀戮尖塔游戏
  • Kubernetes (K8s) 和 Spring Cloud 的区别
  • 定个小目标之刷LeetCode热题(21)
  • Oracle 打开钱包 ORA-28368: cannot auto-create wallet
  • 【麒麟虚拟机】NetworkManager没有运行
  • vue之一键部署的shell脚本和它的点.bat文件、海螺AI、ChatGPT
  • pg和oracle的区别
  • Docker:在DockerHub上创建私有仓库
  • 框架的使用
  • Autosar-DEM诊断事件管理流程