pytorch?tensor合并与分割方式
更新时间：2024年02月20日 09:09:10 作者：wyw0000
这篇文章主要介绍了pytorch?tensor合并与分割方式,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教
（福利推荐：【腾讯云】服务器最新限时优惠活动，云服务器1核2G仅99元/年、2核4G仅768元/3年，立即抢购>>>：9i0i.cn/qcloud）
（福利推荐：你还在原价购买阿里云服务器？现在阿里云0.8折限时抢购活动来啦！4核8G企业云服务器仅2998元/3年，立即抢购>>>：9i0i.cn/aliyun）
1. cat
torch.cat(tensors, dim=0, *, out=None) → Tensor
在指定维度上，连接给定tensor序列或empty，除连接的dimension外，所有得的ensor必须有相同的shape
参数：
- tensors-具有相同类型的tensor序列，非empty tensor必须具有相同的shape，连接的dimension除外
- dim-指定的连接的维度
输出：
连接后的tensor
上图分别是在列和行两个维度连接后的结果
2. stack
创建新维度来连接张量序列
torch.stack(tensors, dim=0, *, out=None) → Tensor
参数：
- tensors-张量序列，具有相同的size
- dim-插入的新维度，必须介于0和连接的tensor的维度之间
输出：
连接后的tensor
注意：cat和stack的区别
stack连接的tensor必须具有相同的size，否则报错，cat是除连接的维度外，其他维度shape必须相同
如下示例：
3. split
把一个tensor切分成块，每个块是原tensor的一部分
torch.split(tensor, split_size_or_sections, dim=0)
参数：
- tensor-用来切分的tensor
- split_size_or_sections (int) or (list(int)) -单个块的size后者是每个块size的list
- dim (int) – 以tensor的哪个维度进行切分
输出：
Tuple[Tensor, …]
示例：
4. chunk
强制将一个tensor切分成指定数量的块，每个块是原tensor的一部分
torch.chunk(input, chunks, dim=0) → List of Tensors
参数：
- input (Tensor) – 输入切分的tensor
- chunks (int) – 切分块的数量
- dim (int) – 以tensor的哪个维度进行切分
输出：
切分后的list
示例：
注意：split与chunk的区别
区别主要是第二个参数，split第二个参数切分块的size，而chunk是切分块的数量
总结
以上为个人经验，希望能给大家一个参考，也希望大家多多支持程序员之家。
相关文章
OpenCV是一个用于计算机视觉和图像操作的免费开源库，有数千种优化的算法和函数用于各种图像操作。本文将使用OpenCV在Python中进行一些图像操作，感兴趣的可以学习一下2022-02-02
这篇文章主要介绍了关于pyqt5控件自适应窗口超详细知识点汇总，有了布局，再在布局中放置各种控件，我们就能让控件实现自适应的效果,需要的朋友可以参考下2023-03-03
这篇文章主要为大家详细介绍了python之Socket网络编程,具有一定的参考价值，感兴趣的小伙伴们可以参考一下2016-09-09
这篇文章主要介绍了Python操作MongoDB数据库的方法,结合实例形式分析了Python命令行模式下操作MongoDB数据库实现连接、查找、删除、排序等相关操作技巧,需要的朋友可以参考下2018-01-01
这篇文章主要介绍了关于python中time和datetime的区别与用法,Python中封装了很多实用的函数,那么关于时间的函数time和datetime有什么区别和作用,让我们来看看吧2023-03-03
这篇文章主要为大家详细介绍了Python如何实现随机生成一个汉字的功能，文中的示例代码讲解详细，对我们深入了解Python有一定的帮助，需要的可以参考一下2023-01-01
这篇文章主要介绍了python读取csv文件指定行的方法详解,需要的朋友可以参考下2020-02-02
这篇文章主要介绍了python根据给定文件返回文件名和扩展名的方法,实例分析了Python操作文件及字符串的技巧,非常具有实用价值,需要的朋友可以参考下2015-03-03
把Anaconda中的环境导入到Pycharm里面的方法步骤
这篇文章主要介绍了把Anaconda中的环境导入到Pycharm里面的方法步骤，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2020-10-10
这篇文章主要介绍了python基础之while循环语句的使用,文中有非常详细的代码示例,对正在学习python的小伙伴们有一定的帮助,需要的朋友可以参考下2021-04-04
最新评论