對于numpy array

當有如下情景，我需要4維tensor，(b,c,w,h), 而寫的程序是單張圖片遍歷的，即每張圖片(c,w,h)，這時，就需要先用一個list，然后每次append，到最后

np.stack(list)

默認是從第0維開始stack

對于tensor

如果我有一個(b,1,w,h)的tensor word_pred，我需要在第二維加一項 1-pred，讓它變成(b,2,w,h)

首先對于(b,c,w,h)的word_pred是可以直接減的，因為有廣播機制

然后

torch.cat((pred_word,1-pred_word),dim=1)

如果用torch.stack的話會給你增加一維?

?

總結

以上是生活随笔為你收集整理的多维tensor/numpy array的拼接(np.stack, torch.cat)的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。