torch.cat()

《动手pytorch》

　　cat ：cconcatenate，拼接

import torch
n_train,n_test = 100,100
features = torch.randn((n_train+n_test))
poly_features = torch.cat((features,torch.pow(features,2),torch.pow(features,3)),1)

　　torch.cat((A,B),dim=0) : 按行拼接，列数不变

 　　torch.cat((A,B),dim=1) : 按列拼接，行数不变