pytorch中stack（）函数和cat（）函数的区别

pytorch框架其中stack（）函数和cat（）函数都是将两个输入组合成一个batch.

stack函数

二者的区别在于，stack（）函数是将两个输入堆叠起来，维度会增加1。

import torch
x = torch.rand((2,2,3))
y = torch.rand((2,2,3))
z = torch.stack((x,y),dim=0)

这样两个堆叠起来的 z 的维度是（2，2，2，3），相当于在dim=0将二者堆叠起来，第一个数字2就是增加的维度。

cat函数

而cat（）函数是将两个输入在现有维度上叠加起来，不改变维度数量。

import torch
x1 = torch.rand((2,2,3))
y1 = torch.rand((2,2,3))
z1 = torch.cat((x,y),dim=0)

这样两个叠加起来的 z1 的维度是（4，2，3），相当于在dim=0将二者叠加起来，不改变输入的维度。

日常学习记录，一起交流讨论吧！侵权联系~