張量的擴展和拼接

 複製 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.],    
相關文章
相關標籤/搜索