张量的扩展和拼接
复制
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)
运行结果: