张量的扩展和拼接

 复制

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)

 运行结果:

猜你喜欢

转载自blog.csdn.net/weixin_42528089/article/details/84839080