pytorch中数据合并与拆分--cat,stack,split,chunk

合并

cat

在这里插入图片描述

要进行cat合并的维度可以不同,但是其余维度必须要相同!

stack

创建一个新的维度,要stack的两个tensor维度要相同!
在这里插入图片描述

用处:将两个班级的学生和成绩合并,生成维度含义:2个表格,每个表格32个学生,每个学生考了8门试!

拆分

split 根据长度拆分

split([2,1],dim=0):在第0维度上,拆分为两个部分,第一部分是2个,第二部分数量是1个。

在这里插入图片描述

chunck 根据数量拆分

在这里插入图片描述
chunk(2,dim=0)在第0维度上,将数据拆分为2块!

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值