pytorch中的cat、stack、tranpose、permute、unsqeeze

pytorch中提供了對tensor經常使用的變換操做。web cat 鏈接 對數據沿着某一維度進行拼接。cat後數據的總維數不變。 好比下面代碼對兩個2維tensor(分別爲2*3,1*3)進行拼接,拼接完後變爲3*3仍是2維的tensor。 代碼以下:svg import torch torch.manual_seed(1) x = torch.randn(2,3) y = torch.ran
相關文章
相關標籤/搜索