expand 就是對那個維度為一的進行擴張

import torcha=torch.tensor([[3.0000, 3.0000],[3.0000, 4.0000],[3.6000, 3.0000],[3.5000, 3.0000]])a1=a.reshape([a.shape[0],a.shape[1] ,1])#a1.shape==[4, 2, 1]#對那個維度為一的進行擴張 a2=a1.expand([4,2,9])print(a2.shape)#torch.Size([4, 2, 9])

總結

以上是生活随笔為你收集整理的torch expand的全部內容，希望文章能夠幫你解決所遇到的問題。

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