import torch

　　

通過 help((torch.cat)) 可以查看 cat 的用法 cat(seq,dim,out=None)其中 seq表示要連接的兩個(gè)序列，以元組的形式給出，例如:seq=(a,b), a,b 為兩個(gè)可以連接的序列 dim 表示以哪個(gè)維度連接，dim=0, 橫向連接dim=1,縱向連接

　

#實(shí)例：#dim=0 時(shí)：import torchn_data = torch.ones((100,2))x0_data = torch.normal(2*n_data,1)y0_data = torch.zeros((100,1))x1_data = torch.normal(-2*n_data,1)y1_data = torch.ones((100,1))x_data = torch.cat((x0_data,x1_data),0).type(torch.FloatTensor)y_data = torch.cat((y0_data,y1_data),0).type(torch.LongTensor)print('x_data的形狀：',x_data.shape)print("y_data的形狀：",y_data.shape)

　　

result:x_data的形狀： torch.Size([200, 2])y_data的形狀： torch.Size([200, 1])

　　

#實(shí)例：#dim=1 時(shí)：import torchn_data = torch.ones((100,2))x0_data = torch.normal(2*n_data,1)y0_data = torch.zeros((100,1))x1_data = torch.normal(-2*n_data,1)y1_data = torch.ones((100,1))x_data = torch.cat((x0_data,x1_data),1).type(torch.FloatTensor)y_data = torch.cat((y0_data,y1_data),1).type(torch.LongTensor)print('x_data的形狀：',x_data.shape)print("y_data的形狀：",y_data.shape)

　　

result：x_data的形狀： torch.Size([100, 4])y_data的形狀： torch.Size([100, 2])

　　

轉(zhuǎn)載于:https://www.cnblogs.com/pythonClub/p/10412418.html

總結(jié)

以上是生活随笔為你收集整理的Pytorch的torch.cat实例的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。