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

这一节,将分析拼接(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教程(十九)torch.cat与torch.stack的区别_第1张图片
在pytorch中,我们通过stack和cat进行操作:
Pytorch教程(十九)torch.cat与torch.stack的区别_第2张图片

你可能感兴趣的:(pytorch教程,pytorch,深度学习,tensorflow)