stack()
拼接函数,是拼接以后,再扩展一维。torch.stack()有两个参数,第一个是要拼接的张量的列表或是元组;第二个参数是拼接的维度。
cat()
拼接函数,将多个张量拼接成一个张量,保持维度不变。torch.cat()有两个参数,第一个是要拼接的张量的列表或是元组;第二个参数是拼接的维度。
torch.cat() 和 torch.stack()函数要求输入张量的大小完全相同,才能进行拼接
stack与cat的区别在于, 得到的张量的维度会比输入的张量的大小多1,并且多出的那个维度就是拼接的维度,那个维度的大小就是输入张量的个数。
A=torch.tensor([[1,2,3],[4,5
PyTorch:torch.cat()与torch.stack()函数详解及应用

最低0.47元/天 解锁文章
10万+

被折叠的 条评论
为什么被折叠?



