1. torch.view()

data.view(a,b) # 把data 轉換成 a*b 維的矩陣 data.view(k,-1)

2. torch.squeeze()

data.squeeze(0) # 表示在第一維度壓縮，如果第一維度是1，則直接去掉 img=content img.shape torch.Size([1，3, 800, 600]) img=content.squeeze(0) img.shape torch.Size([3, 800, 600])

3. torch.unsqueeze()

與sequeeze() 相反，增加一個維度，維度為1

總結

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

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