torch.cat与torch.stack的区别

最新推荐文章于 2024-08-20 13:09:25 发布

沐风大大

最新推荐文章于 2024-08-20 13:09:25 发布

阅读量969

点赞数 4

分类专栏： PyTorch之函数文章标签： stack pytorch

本文链接：https://blog.csdn.net/weixin_42621901/article/details/109103604

版权

本文介绍了PyTorch中torch.cat()和torch.stack()的用法。torch.cat()用于在指定维度上拼接张量，而torch.stack()则会在张量的维度之外新增一维进行拼接。通过实例解析了两者在操作上的差异。

摘要由CSDN通过智能技术生成

torch.cat()的用法

torch.cat用于矢量的拼接，例如：

# torch.cat用法：
import torch
A = torch.ones(2,3)
print(A)
B = torch.ones(2,3)
print(B)
C = torch.cat((A,B),0) # 按行拼接
print(C)
D = torch.cat((A,B),1) # 按列拼接
print(D)
------------输出----------------------
tensor([[1., 1., 1.],
        [1., 1., 1.]