PyTorch cat PyTorch的concat也就是torch.cat实例

PyTorch的concat也就是torch.cat实例

import torch
a = torch.ones([1,2])
b = torch.ones([1,2])
torch.cat([a,b],1)
 1  1  1  1
[torch.FloatTensor of size 1x4]