这一节,将分析拼接(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进行操作: