Pytorch torch

Pytorch torch.cat函数

cat函数将两个张量拼接在一起.

如下例子:

import torch
a=torch.ones([1,3])
b=torch.ones([1,3])
torch.cat([a,b],1) #按列拼接
#输出:tensor([[ 1.,  1.,  1.,  1.,  1.,  1.]])
torch.cat([a,b],0) #按行拼接
#输出:tensor([[ 1.,  1.,  1.],
#        [ 1.,  1.,  1.]])