张量的扩展和拼接

 复制

t12= torch.tensor([[5.,-9.],])#大小（3,4）
print('t12={}'.format(t12))
t34= t12.repeat(3,2)#三行，每一行复制2次
print('t34={}'.format(t34))

运行结果：

t12=tensor([[ 5., -9.]])
t34=tensor([[ 5., -9.,  5., -9.],
        [ 5., -9.,  5., -9.],
        [ 5., -9.,  5., -9.]])

拼接1：

tp=torch.arange(12).reshape(3,4)
print(tp)
tn=-tp
tc0=torch.cat([tp,tn],1)#行数不增加，列数增加
print(tc0)
tc0=torch.cat([tp,tn],0)#列数不增加，行数增加
print(tc0)

拼接2：

 

tp=torch.arange(12).reshape(3,4)
print(tp)
tn=-tp
tc0=torch.stack([tp,tn],1)#行数增加，列数不增加，且交叉
print(tc0)
tc0=torch.stack([tp,tn],0)#列数不增加，行数增加
print(tc0)

 运行结果：