PyTorch的torch.cat用法

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