秀丽抽屉

文章
6
资源
0
加入时间
2年10月24天

pytorch中stack()函数和cat()函数的区别

pytorch框架其中stack()函数和cat()函数都是将两个输入组合成一个batch.stack函数二者的区别在于,stack()函数是将两个输入堆叠起来,维度会增加1。import torchx = 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()函数是将