pytorch tensor合并与分割方式
更新时间：2024年02月20日 09:09:10 作者：wyw0000
这篇文章主要介绍了pytorch tensor合并与分割方式,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教
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是切分块的数量
总结
以上为个人经验，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
这篇文章给大家介绍的一段脚本是自动添加注释信息的脚本，添加的信息包括脚本名称、作者和时间等之类的，对于团队形成统一的编码规则很有帮助。有需要的可以参考借鉴。2016-09-09
这篇文章主要介绍了python自动化测试之setUp与tearDown实例,其中setUp()方法中进行测试前的初始化工作,并在tearDown()方法中执行测试后的清除工作,setUp()和tearDown()都是TestCase类中定义的方法,需要的朋友可以参考下2014-09-09
这篇文章主要介绍了pandas to_excel 添加颜色操作，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2020-07-07
Django提供了一套非常方便的类似SqlAlchemy ORM的通过对象调用的方式操作数据库表的ORM框架,,本文给大家详细介绍Django ORM查询操作方式,感兴趣的朋友一起看看吧2023-10-10
Python 中 Pandas 文件操作和读取 CSV 参数详解
CSV 又称逗号分隔值文件，是一种简单的文件格式，以特定的结构来排列表格数据，这篇文章主要介绍了Python 之 Pandas 文件操作和读取 CSV 参数详解,需要的朋友可以参考下2023-03-03
这篇文章主要介绍了python记录程序运行时间的三种方法的相关资料,需要的朋友可以参考下2017-07-07
这篇文章主要介绍了python列表和字符串的三种逆序遍历操作，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教2021-06-06
这篇文章主要讲解得内容是python与js进行MD5取hash有什么不同，我们在做前端做渗透测试时会遇到一些关键字进行了加密得情况，而且python和js对json进行md5取hash,MD5结果值还不一致，下面我们就烂看看到底是哪里不同吧，需要的朋友可以参考一下2022-02-02
今天小编就为大家分享一篇python 实现保存最新的三份文件,其余的都删掉，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2019-12-12
这篇文章主要介绍了python判断图片相似度找出相似的图像问题,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教2026-03-03
最新评论