Pytorch常用函數

1.view() 直接上例子： import torch x=torch.randn(6,6) y=x.view(x.size(0),2,-1)  #保持第一維，對後面兩維做resize y.size()=(6,2,3) 2.cat (多個tensors的拼接) import torch x=torch.randn(2,3) torch.cat((x,x),0)  #行拼接4*3 torch.ca

>>阅读原文<<