你的位置:首页 > 信息动态 > 新闻中心
信息动态
联系我们

Pytorch教程(十九)torch.cat与torch.stack的区别

2021-12-5 18:05:27

这一节,将分析拼接(concatenating)和叠加(stacking)张量的区别。

首先看三个例子,一个是pytorch,一个是TensorFlow,一个是numpy。

stack和cat张量之间的区别可以用一句话描述:

Concatenating joins a sequence of tensors along an existing axis, and Stacking joins a sequence of tensors along a new axis.

也就是说cat是在已有的轴上进行连接,而stack是创建一个新轴,进行连接。

在这里插入图片描述
在pytorch中,我们通过stack和cat进行操作:
在这里插入图片描述