1、torch.cat( (A,B),dim=i )
第i维保持不变拼接AB
2、torch.chunk(tensorA, chunk=i, dim=j )
将张量A沿着j轴分为i部分（不能整除最后一部分缺）
3、assert
检查条件，不满足报错
扫一扫
评论
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
查看更多评论
添加红包
1、torch.cat( (A,B),dim=i )
第i维保持不变拼接AB
2、torch.chunk(tensorA, chunk=i, dim=j )
将张量A沿着j轴分为i部分（不能整除最后一部分缺）
3、assert
检查条件，不满足报错
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
