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

62 阅读 0 评论 41 点赞

我是靠谱客的博主秀丽抽屉，最近开发中收集的这篇文章主要介绍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（）函数是将两个输入在现有维度上叠加起来，不改变维度数量。

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将二者叠加起来，不改变输入的维度。

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

如果觉得靠谱客网站的内容还不错，欢迎将靠谱客网站推荐给程序员好友。

本图文内容来源于网友提供，作为学习参考使用，或来自网络收集整理，版权属于原作者所有。