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中，我们通过stack和cat进行操作：
在这里插入图片描述

版权声明：本文为vincent_duan原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接和本声明。