分类：torch.cat

2021
07-08

聊聊Pytorch torch.cat与torch.stack的区别

torch.cat()函数可以将多个张量拼接成一个张量。torch.cat()有两个参数，第一个是要拼接的张量的列表或是元组；第二个参数是拼接的维度。torch.cat()的示例如下图1所示图1torch.cat()torch.stack()函数同样有张量列表和维度两个参数。stack与cat的区别在于，torch.stack()函数要求输入张量的大小完全相同，得到的张量的维度会比输入的张量的大小多1，并且多出的那个维度就是拼接的维度，那个维度的大小就是输入张量的个数。torch.... 继续阅读 >

2020
10-08

PyTorch的torch.cat用法

1.字面理解：torch.cat是将两个张量（tensor）拼接在一起，cat是concatnate的意思，即拼接，联系在一起。2.例子理解>>>importtorch>>>A=torch.ones(2,3)#2x3的张量（矩阵）>>>Atensor([[1.,1.,1.],[1.,1.,1.]])>>>B=2*torch.ones(4,3)#4x3的张量（矩阵）>>>Btensor([[2.,2.,2.],[2.,2.,2.],[2.,2.,2.],[2.,2.,2.]])>>>C=torch.cat((A,B),0)#按... 继续阅读 >