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是切分块的数量
总结
以上为个人经验，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
Django-simple-captcha验证码包使用方法详解
这篇文章主要介绍了Django-simple-captcha验证码包使用方法详解,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2020-11-11
使用pip installopencv-python安装时,安装速度很慢,本文主要介绍了Python安装OpenCV库超时失败,具有一定的参考价值,感兴趣的可以了解一下2024-05-05
这篇文章主要为大家详细介绍了python实现梯度下降和逻辑回归，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2020-03-03
这篇文章主要为大家介绍了Python数据结构集合set教程示例详解,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪2023-09-09
django2用iframe标签完成网页内嵌播放b站视频功能
这篇文章主要介绍了django2 用iframe标签完成 网页内嵌播放b站视频功能,小编觉得挺不错的，现在分享给大家，也给大家做个参考。一起跟随小编过来看看吧2018-06-06
社区版pycharm创建django项目的方法(pycharm的newproject左侧没有项目选项)
pycharm的newproject左侧没有出现项目选项的情况下，创建Django项目的解决方法./社区版pycharm创建django项目的方法，本文给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友参考下吧2020-09-09
这篇文章主要介绍了Pytorch 使用CNN图像分类的实现，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2020-06-06
python基于scrapy爬取京东笔记本电脑数据并进行简单处理和分析
这篇文章主要介绍了python基于scrapy爬取京东笔记本电脑数据并进行简单处理和分析的实例，帮助大家更好的理解和学习使用python。感兴趣的朋友可以了解下2021-04-04
这篇文章主要介绍了关于自动化测试框架pytest的Fixture固件,Fixture它其实就是一些函数,会在执行测试方法/测试函数前后加载运行它们,需要的朋友可以参考下2023-03-03
这篇文章主要介绍了Python实现读取Properties配置文件的方法,结合实例形式分析了Python读取Properties配置文件类的定义与使用相关操作技巧,需要的朋友可以参考下2018-03-03
最新评论