Pytorch中的torch.cat()函數

cat是concatnate的意思：拼接，聯繫在一起。   先說cat( )的普通用法 如果我們有兩個tensor是A和B，想把他們拼接在一起，需要如下操作： C = torch.cat( (A,B),0 ) #按維數0拼接（豎着拼） C = torch.cat( (A,B),1 ) #按維數1拼接（橫着拼）   >>> import torch >>> A=torch.ones(2,3)

>>阅读原文<<